算法通关村第七关 | 迭代遍历二叉树

        迭代是重复反馈过程的活动，其目的通常是为了逼近所需目标或结果。每一次对过程的重复称为一次“迭代”，而每一次迭代得到的结果会作为下一次迭代的初始值。

1.迭代法实现前序遍历

前序遍历是中左右，如果还有左子树就一直向下找，到最低之后返回逐步向上向右找，

代码：

    //迭代法实现前序遍历
    public List<Integer> preOrderTraversal(TreeNode root){
        List<Integer> res = new ArrayList<>();
        if (root == null){
            return res;
        }
        Deque<TreeNode> stack = new LinkedList<>();
        TreeNode node = root;
        while (!stack.isEmpty() || node != null){
            //前序根左右
            while (node != null) {
                res.add(node.val);
                stack.push(node);
                node = node.left;
            }
            //到最左的时候开始弹出向上向右找
            node = stack.pop();
            node = node.right;
        }
        return res;
    }

2.迭代法实现中序遍历

中序遍历是左中右，类似于前序遍历，遍历出来的是左，只需和前序遍历稍微调动一下位置，

    //迭代法实现中序遍历
    public List<Integer> midOrderTraversal(TreeNode root){
        List<Integer> res = new ArrayList<>();
        Deque<TreeNode> stack = new LinkedList<>();
        while (!stack.isEmpty() || root != null){
            //前序根左右
            while (root != null) {
                stack.push(root);
                root = root.left;
            }
            //到最左的时候开始弹出向上向右找
            root = stack.pop();
            res.add(root.val);
            root = root.right;
        }
        return res;
    }

3.迭代法实现后序遍历

反转法实现后序遍历。

 

如下图，我们先观察后序遍历的结果是 order={9 5 7 4 3}，将其反转，new_order={3 4 7 5 9}。

前序遍历{3 9 4 5 7}，前序是先中间再左边再右边，而这里的反转是先中间再右边再左边，我们只需改动前序遍历，得到后序的反转，再将其反转得到后序的遍历。

    public List<Integer> endOrderTraversal(TreeNode root){
        List<Integer> res = new ArrayList<>();
        if (root == null){
            return res;
        }
        Deque<TreeNode> stack = new LinkedList<>();
        TreeNode node = root;
        while (!stack.isEmpty() || node != null){
            while (node != null) {
                res.add(node.val);
                stack.push(node);
                node = node.right;
            }
            //到最右的时候开始弹出向上向左找
            node = stack.pop();
            node = node.left;
        }
        Collections.reverse(res);
        return res;
    }