算法通关村—迭代实现二叉树的遍历

68 阅读1分钟

递归能做的迭代一定能做,只不过递归会复杂一些

迭代法实现前序遍历

144. 二叉树的前序遍历

题目:给你二叉树的根节点 root ,返回它节点值的 前序 遍历。

图示

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<Integer>();
        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;
    }
}

迭代法实现中序遍历

把 res 添加的位置调一下就可以了,要是看不懂的话应该是上面的没理解。

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

迭代法实现后序遍历

总的来所右这么几种方法

  1. 反转法
  2. 访问标记法
  3. Morris法: 链接:zhuanlan.zhihu.com/p/101321696

我们重点讲 反转方

我们观察到,后续排序反转之后就变成了类似于前序排序的算法

    public  List<Integer> postOrderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        if (root == null) return res;
        Stack<TreeNode> stack = new Stack<>();
        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;
    }