cubegao

Swift.二叉树的前序遍历

2015-05-24

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import Foundation

class MyTreeNodeSolution {

//递归
func preorderTraversal(_ root: TreeNode?,_ n: inout [Int]) {

if root == nil {
return
}

n.append(root!.val)
preorderTraversal(root?.left, &n)
preorderTraversal(root?.right, &n)
}

//非递归
func preorderTraversal2(_ root: TreeNode?) -> [Int] {

var node = root
var stack = [TreeNode?]()
var res = [Int]()

while !stack.isEmpty || node != nil {

if node != nil {
res.append(node!.val)
stack.append(node)
node = node?.left
}else {
node = stack.removeLast()?.right
}
}

return res
}
}

算法思想:前序遍历:根结点 —> 左子树 —> 右子树

github地址:https://github.com/cubegao/LeetCode

Tags: 算法

扫描二维码,分享此文章