Skip to content

Instantly share code, notes, and snippets.

@AndyBowes
AndyBowes / TreeNode.kt
Created August 2, 2017 11:20
Kotlin TreeNode (Draft)
fun idSequence(seed:Int=0) : Sequence<Int>{
return generateSequence(seed,{it + 1})
}
class TreeNode<T>(val id: T){
val childNodes = mutableListOf<TreeNode<T>>()
fun addChild(node: TreeNode<T>) = childNodes.add(node)
fun removeChild(node: TreeNode<T>) = childNodes.remove(node)
fun children() = childNodes.asIterable()