주어진 깊이에서 노드의 스위프트 이진 트리 목록

Question

안녕하세요 저는 이진 트리에 대한 신속한 알고리즘을 작성하고 있습니다. 내 목표는

여기

func listNodeAt(_n: Int) --> [T] { 

} 

내 나무 클래스

public class BinaryTreeNode<T:Comparable> { 

    //Value and children vars 
    public var value:T 
    public var leftChild:BinaryTreeNode? 
    public var rightChild:BinaryTreeNode? 
    public weak var parent:BinaryTreeNode? 

    //Initialization 
    public convenience init(value: T) { 
     self.init(value: value, left: nil, right: nil, parent:nil) 
    } 

    public init(value:T, left:BinaryTreeNode?, right:BinaryTreeNode?, parent:BinaryTreeNode?) { 
     self.value = value 
     self.leftChild = left 
     self.rightChild = right 
     self.parent = parent 
    } 
} 

처럼 내가 노드의 깊이를 계산하는 도우미 함수를 구축해야 특정 깊이에서 뭔가를 노드의 목록을 만드는 것입니다

//Depth 
    public func depth() -> Int { 
     guard var node = parent else { 
      return 0 
     } 

     var depth = 1 
     while let parent = node.parent { 
      depth = depth + 1 
      node = parent 
     } 

     return depth 
    } 

어떻게하면 원하는 기능을 얻을 수 있습니까? 어떤 제안이라도 대단히 감사합니다. 감사!

Answer 1

func listNodeAt(_ n: Int) -> [T] { 
    return getElementsAt(n, node: self) 
} 

private func getElementsAt(_ n: Int, node: BinaryTreeNode<T>, traversingDepth: Int = 0) -> [T] { 
     var array = Array<T>() 
     if traversingDepth < n { 
      if let left = node.leftChild { 
       array = array + getElementsAt(n, node: left, traversingDepth: traversingDepth + 1) 
      } 
      if let right = node.rightChild { 
       array = array + getElementsAt(n, node: right, traversingDepth: traversingDepth + 1) 
      } 
     } else if traversingDepth == n { 
      array.append(node.value) 
     } 
     return array 
    } 

이것은 해결책 중 하나입니다. 여기서 자기가 루트 노드라고 가정합니다.