CodeTop 刷题笔记 102. 二叉树的层序遍历

最新推荐文章于 2024-10-09 22:28:58 发布

余生!

最新推荐文章于 2024-10-09 22:28:58 发布

阅读量94

点赞数

分类专栏：算法文章标签：深度优先算法 leetcode 广度优先

本文链接：https://blog.csdn.net/qq_16162595/article/details/125340001

版权

算法专栏收录该内容

1 篇文章 0 订阅

订阅专栏

题目

给你二叉树的根节点 root ，返回其节点值的层序遍历。（即逐层地，从左到右访问所有节点）。
在这里插入图片描述

思路

BFS 广度优先遍历

借用队列

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();

        if (root == null){
            return res;
        }
        Queue<TreeNode> que = new LinkedList<>();
        que.offer(root);

        while(! que.isEmpty()){
            List<Integer> item = new ArrayList<>();
            int size = que.size();
            while(size>0){
                TreeNode temp = que.poll();
                if (temp.left != null) {
                    que.offer(temp.left);
                }
                if (temp.right != null){
                    que.offer(temp.right);
                }
                item.add(temp.val);
                size--;
            }
            res.add(item);
        }
        return res;
    }
}

DFS 深度优先遍历

class Solution {
    public List<List<Integer>> res = new ArrayList<>();
    public List<List<Integer>> levelOrder(TreeNode root) {
        dfs(root, 0);
        return res;
    }

    public void dfs(TreeNode node, Integer deep){
        if (node == null) return;

        deep++;
        if(res.size() < deep){
            //deep 表示层数
            List<Integer> item = new ArrayList<>();
            res.add(item);
        }
        res.get(deep-1).add(node.val);//直接根据索引加到对应的位置上
        dfs(node.left, deep);
        dfs(node.right, deep);
    }
}