代码随想录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;
    }
}

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

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;
    }
}

  • 5
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值