代码随想录day13 Java版 二叉树部分-遍历全解

1.二叉树的递归遍历

回顾了递归的套路：1.返回值+传入参数

                                   2.结束条件

                                   3.写具体的每一个递归单位

而在二叉树的三种遍历都没有返回值，传入当前node节点和所给框架下的返回值数组res

当传入的节点是null时结束递归

每个递归单位包含输出当前节点+递归左孩子+递归右孩子，改变输出的位置即可实现三种遍历

144.二叉树的前序遍历

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        pre(root,res);
        return res;
    }
    void pre(TreeNode node, List<Integer> res){
        if(node == null) return;
        res.add(node.val);
        pre(node.left,res);
        pre(node.right,res);
    }
}

145.二叉树的后序遍历

class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        post(root,res);
        return res;
    }
    void post(TreeNode node, List<Integer> res){
        if(node == null) return;
        post(node.left,res);
        post(node.right,res);
        res.add(node.val);
    }
}

94.二叉树的中序遍历

class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        in(root,res);
        return res;
    }
    void in(TreeNode node, List<Integer> res){
        if(node == null) return;
        in(node.left,res);
        res.add(node.val);
        in(node.right,res);
    }
}

2.二叉树的迭代遍历

144.二叉树的前序遍历

众所周知递归都可以改成循环，本质上还是自己模拟递归栈的过程

前序遍历模拟最简单，因为处理的节点和先访问的节点一致，可以直接将节点的左右孩子入栈

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        if (root == null) return res;
        Deque<TreeNode> s = new LinkedList<>();
        s.push(root);
        while (!s.isEmpty()) {
            TreeNode cur = s.pop();
            res.add(cur.val);
            if (cur.right != null) s.push(cur.right);
            if (cur.left != null)  s.push(cur.left);
        }
        return res;
    }
}

• 145.二叉树的后序遍历(opens new window)

稍微改动下前序遍历的代码，左右孩子改下顺序再将最后的结果逆序就可以得到后序遍历的结果

Java里的集合进行逆序：Collections.reverse(res)

class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        if (root == null) return res;
        Deque<TreeNode> s = new LinkedList<>();
        s.push(root);
        while (!s.isEmpty()) {
            TreeNode cur = s.pop();
            res.add(cur.val);
            if (cur.left != null)  s.push(cur.left);
            if (cur.right != null) s.push(cur.right);
        }
        Collections.reverse(res);
        return res;
    }
}

94.二叉树的中序遍历

中序遍历没办法像前序遍历直接模拟，需要一路向左入栈，左孩子为空后处理，再查看右孩子

代码上while内的条件直接作为两个分支处理

class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> res = new LinkedList<>();
        if (root == null) return res;
        Deque<TreeNode> s = new LinkedList<>();
        TreeNode cur = root;
        while(cur != null || !s.isEmpty()){
            if (cur != null) {
                s.push(cur);
                cur = cur.left;
            }else {
                cur = s.pop();
                res.add(cur.val);
                cur = cur.right;
            }
        }
        return res;
    }
}

3.二叉树的层序遍历

按照广度优先搜索，从根节点（第一层）开始，输出节点后将节点的左右孩子入队

注意代码中队列使用offer和poll，以及容器的size要在循环前保存

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        if (root == null) return res;
        Deque<TreeNode> q =new LinkedList<>();
        q.offer(root);
        while(!q.isEmpty()){
            List<Integer> list=new ArrayList<>();
            int n = q.size();
            for (int i = 0; i < n; i++) {
                TreeNode node = q.poll();
                list.add(node.val);
                if (node.left != null)  q.offer(node.left);
                if (node.right != null) q.offer(node.right);
            }
            res.add(list);
        }
        return res;
    }
}