LeetCode102:二叉树的层序遍历

给你二叉树的根节点 root ,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。

示例:
输入:root = [3,9,20,null,null,15,7]
输出:[[3],[9,20],[15,7]]

广度优先遍历是按层层推进的方式,遍历每一层的节点。题目要求的是返回每一层的节点值,所以这题用广度优先来做非常合适。
广度优先需要用队列作为辅助结构,我们先将根节点放到队列中,然后不断遍历队列。

首先拿出根节点,如果左子树/右子树不为空,就将他们放入队列中。根节点放入之后,根节点再从队列中拿走了,而根节点的两个孩子放入队列中
我们把每层遍历到的节点都放入到一个结果集中,最后返回这个结果集就可以了。
时间复杂度: O(n)
空间复杂度:O(n)

public List<List<Integer>> levelOrder(TreeNode root) {
    if(root == null){
        return new ArrayList<List<Integer>>();
    }

    List<List<Integer>> res = new ArrayList<List<Integer>>();
    LinkedList<TreeNode> queue = new LinkedList<>();
    //先将根节点放入队列中
    queue.add(root);
    while (queue.size() > 0){
        //获取当前队列的长度,即当前层的节点数
        int size = queue.size();
        ArrayList<Integer> tmp = new ArrayList<>();
        for (int i = 0; i < size; i++) {
        	//将队列中的元素拿出来(这一层的节点),移除的元素加入到临时集合tmp中
            TreeNode remove = queue.remove();
            tmp.add(remove.val);
            //如果节点的左右子树不空,放入队列中,以便下次循环加入到临时集合中
            if (remove.left != null){
                queue.add(remove.left);
            }
            if (remove.right != null){
                queue.add(remove.right);
            }
        }
        //将临时集合tmp加入到结果集中
        res.add(tmp);
    }
    return res;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值