swift - 二叉树的遍历

最常见的二叉树遍历有三种

前序遍历：先访问根节点——左子树——右子树。
中序遍历：先访问左子树——根节点——右子树，按照这个顺序。
后序遍历：和前面差不多，先访问树的左子树——右子树——根节点。按层遍历：把一棵树从上到下，从左到右依次写出来。

例子：

在这里插入图片描述

一栈的前序遍历

答案：
在这里插入图片描述

思路：第一步：从根节点访问
第二步：只要节点有值，访问完放入栈中，把值存入数据，
第三步：把左节点的值复制给节点，依次循环第二步
第四步：当左节点复制完节点之后为nil 代表左边没值了
第五步：当左边没值，移除栈的最后一个，并把移除后的栈的右节点复制给节点，重复这几步

代码：

 func p_preorderTravelsal(root : TreeNode?) -> [Int] {
        
        var res = [Int]()
        var stack = [TreeNode]()
        var node = root
        
        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
    }

测试：

 @objc func preorderTravelsal()
    {
        //给定二叉树 [28, 16, 30, 13, 22,29，43]
        
        let root = TreeNode.init(val: 28)
        root.left = TreeNode.init(val: 16)
        root.right = TreeNode.init(val: 30)
        
        let second = root.left
        second?.left = TreeNode.init(val: 13)
        second?.right = TreeNode.init(val: 22)
        
        let thred = root.right
        thred?.left = TreeNode.init(val: 29)
        thred?.right = TreeNode.init(val: 43);
        
        let result = self.p_preorderTravelsal(root: root)
        print("前序遍历的结果\(result)")
    }

结果：

在这里插入图片描述

一栈的中序遍历

答案：
在这里插入图片描述

思路：第一步：从根节点开始访问
第二步：访问完放入栈中，把节点的左节点复制给节点，
第三步：当节点没有值时，把栈的最后一个值得节点放入数组中，然后移除最后一个栈，拿到倒数第二个，把值得右节点赋值给节点，循环第二步，第三步。

代码：

//栈的中序遍历
    func p_inorderTravelsal(root : TreeNode?) -> [Int] {
        
        var res = [Int]()
        var stack = [TreeNode]()
        var node = root
        
        while !stack.isEmpty || node != nil {
            if node != nil {
                
                stack.append(node!)
                node = node!.left
                
            }else{
                
                res.append(stack.last!.val)
                node =  stack.removeLast().right

            }
        }
        return res
    }

测试：

@objc func inorderTravelsal()
    {
        //给定二叉树 [28, 16, 30, 13, 22,29，43]
        
       // 结果：[13,16,22,28,29,30,43]
        let root = TreeNode.init(val: 28)
        root.left = TreeNode.init(val: 16)
        root.right = TreeNode.init(val: 30)
        
        let second = root.left
        second?.left = TreeNode.init(val: 13)
        second?.right = TreeNode.init(val: 22)
        
        let thred = root.right
        thred?.left = TreeNode.init(val: 29)
        thred?.right = TreeNode.init(val: 43);
        
        let result = self.p_inorderTravelsal(root: root)
        print("中序遍历的结果\(result)")
    }