12.29

二叉树的迭代遍历

总结

总体思路是通过栈的入栈出栈顺序来控制输出结果,也是用栈来模拟递归。

如前序遍历，中左右，则根节点，入栈后立即出栈，加入右左两节点。左节点成为下一个根节点。

中序遍历，比较麻烦，左中右，则事先根节点入栈，并不断向左节点探寻，直到叶子节点，然后按顺序出栈中的根节点，每出一个，则继续探寻其右节点是否存在，把右节点当做根节点继续向其左节点探寻。

后序遍历，左右中，事先按照前序遍历的方法反转遍历中右左，时候进行反转。
中序比较麻烦，可以多理解理解。
但是这样迭代中序跟其他两种不一样，写的时候比较麻烦。于是给出了统一解法，打上标记。但个人感觉没什么，理解起来比较麻烦。熟练掌握普通的迭代方法就行。

代码

    public static List<Integer> preorderTraversal(TreeNode root){
        Stack<TreeNode> tr=new Stack<>();
        List<Integer> res=new ArrayList<>();
        tr.push(root);
        while (!tr.isEmpty()){
            TreeNode tempNode=tr.pop();
            res.add(tempNode.val);
            if (tempNode.right!=null)tr.push(tempNode.right);
            if (tempNode.left!=null) tr.push(tempNode.left);
        }
        return res;
    }

    public static List<Integer> inorderTraversal(TreeNode root){
        Stack<TreeNode> stack=new Stack<>();
        List<Integer> res=new ArrayList<>();
        stack.push(root);
        TreeNode tempNode=root;
        while (tempNode != null || !stack.isEmpty()){
            if (tempNode != null){
                stack.push(tempNode);
                tempNode = tempNode.left;
            }else{
                tempNode = stack.pop();
                res.add(tempNode.val);
                tempNode = tempNode.right;
            }
        }
        return res;
    }

    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();


        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        while (!stack.isEmpty()){
            TreeNode tempNode = stack.pop();
            res.add(tempNode.val);
            if (tempNode.left != null){
                stack.push(tempNode.left);
            }
            if (tempNode.right != null){
                stack.push(tempNode.right);
            }
        }
        Collections.reverse(res);
        return res;
    }