【二叉树】二叉树的前序遍历

0x00 遍历方式

二叉树的遍历方式：

• 深度优先
• 广度优先

深度优先：

• 前序遍历：中左右
• 中序遍历：左中右
• 后序遍历：左右中

广度优先：

• 层序遍历：遍历完同一层的所有节点后，继续往下层遍历

---

0x01 前序遍历

语言：Swift

树节点：TreeNode

public class TreeNode {
    public var val: Int
    public var left: TreeNode?
    public var right: TreeNode?
    public init() { self.val = 0; self.left = nil; self.right = nil; }
    public init(_ val: Int) { self.val = val; self.left = nil; self.right = nil; }
    public init(_ val: Int, _ left: TreeNode?, _ right: TreeNode?) {
        self.val = val
        self.left = left
        self.right = right
    }
}

递归遍历：

func preorder(_ root: TreeNode?) -> [Int] {
    guard let root = root else { return [] }
    
    var array = [Int]()
    func dfs(_ root: TreeNode?) {
        guard let root = root else { return }
        array.append(root.val) // 中
        dfs(root.left)		   // 左
        dfs(root.right)	       // 右
    }
    
    dfs(root)
    return array
}

迭代遍历：
需要手动维护一个栈结构，而递归方式有函数调用栈

func preorder(_ root: TreeNode?) -> [Int] {
    guard let root = root else { return [] }
    
    var array = [Int]()
    var stack = [TreeNode]()
    var t: TreeNode? = root
    
    while t != nil || !stack.isEmpty {
        while t != nil { // 先左，一直到最下边
            stack.append(t!)  // 入栈
            array.append(t!.val)           // 中
            t = t!.left //再下一个          // 左
        }
        // 再切换到右边
        t = stack.removeLast().right      // 右
    }
    return array
}

---