RediOne1/Tree-Data-Structure
1
0
Fork 0
mirror of https://github.com/AdrianKuta/Tree-Data-Structure.git synced 2025-10-16 18:22:30 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

24: Add iterator param to TreeNode's constructor. (#25)

Browse Source
This commit is contained in:
Adrian Kuta
2023-07-24 18:50:46 +02:00
committed by GitHub
parent d34050e9af
commit 04c3728fcd
5 changed files with 10 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/commonMain/kotlin/com.github.adriankuta/datastructure/tree/TreeNode.kt
View File

@@ -8,7 +8,10 @@ import com.github.adriankuta.datastructure.tree.iterators.TreeNodeIterators
import com.github.adriankuta.datastructure.tree.iterators.TreeNodeIterators.*
import kotlin.jvm.JvmSynthetic
open class TreeNode<T>(val value: T) : Iterable<TreeNode<T>>, ChildDeclarationInterface<T> {
/**
* @param treeIterator Choose one of available iterators from [TreeNodeIterators]
*/
open class TreeNode<T>(val value: T, var treeIterator: TreeNodeIterators = PreOrder) : Iterable<TreeNode<T>>, ChildDeclarationInterface<T> {
private var _parent: TreeNode<T>? = null
@@ -33,11 +36,6 @@ open class TreeNode<T>(val value: T) : Iterable<TreeNode<T>>, ChildDeclarationIn
val isRoot: Boolean
get() = _parent == null
/**
* Choose one of available iterators from [TreeNodeIterators]
*/
var defaultIterator: TreeNodeIterators = PreOrder
/**
* Add new child to current node or root.
*
@@ -166,9 +164,9 @@ open class TreeNode<T>(val value: T) : Iterable<TreeNode<T>>, ChildDeclarationIn
}
/**
* You can change default iterator by changing [defaultIterator] property.
* You can change default iterator by changing [treeIterator] property.
*/
override fun iterator(): Iterator<TreeNode<T>> = when (defaultIterator) {
override fun iterator(): Iterator<TreeNode<T>> = when (treeIterator) {
PreOrder -> PreOrderTreeIterator(this)
PostOrder -> PostOrderTreeIterator(this)
LevelOrder -> LevelOrderTreeIterator(this)

3
src/commonMain/kotlin/com.github.adriankuta/datastructure/tree/TreeNodeExt.kt
View File

@@ -19,8 +19,7 @@ inline fun <reified T> tree(
defaultIterator: TreeNodeIterators = TreeNodeIterators.PreOrder,
childDeclaration: ChildDeclaration<T>
): TreeNode<T> {
val treeNode = TreeNode(root)
treeNode.defaultIterator = defaultIterator
val treeNode = TreeNode(root, defaultIterator)
treeNode.childDeclaration()
return treeNode
}