LeetCode102. 二叉树的层序遍历

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

以下是个人做这道题时的一些理解

解题思路:题目要求层序遍历二叉树,输出结果是可以区分每一层的元素


        我们使用广度优先遍历(BFS)输出的结果是[3 9 20 15 7],可见与层序遍历输出结果是一样的,但是题目要求层序遍历可以区分开每一层的遍历结果,如:[[3],[9,20],[15,7]],要如何处理呢?

首先我们要了解广度优先遍历(BFS)是怎么遍历二叉树的:

        实现广度优先我们借助了队列这一数据结构,先将要遍历的二叉树的根节点添加到队列,为实现BFS做准备条件,然后在出队列与入队列的循环迭代中,来实现层序遍历

层序遍历的某个过程:

 

        

        可见在第一层的3节点还未完全出栈,节点2的左右节点(第三层)已经入队列了,这样我们在出队列时就无法使每一层的元素分开,所以,在每一次出队列之前,我们应该获取到此时队列的长度length,每次出队列的次数就是length次,这样就能保证一次性把同一层的所有元素出队列完毕后不在出队列,从而混入下一层的元素, 这样就实现了每层元素分离的效果。

完整代码如下:

public List<List<Integer>> levelOrder(TreeNode root) {
    //先定义一个存放整个层序遍历结果的list
    List<List<Integer>> res = new ArrayList<>();
    if(root == null){
        return new ArrayList<>();
}
    //定义队列
    Queue<TreeNode> queue = new LinkedList<TreeNode>();
    //将root节点入队列,为BFS做准备
    queue.add(root);
    while(!queue.isEmpty()){
        //获取每次出队列之前的队列长度
        int count = queue.size();
 //定义list用来存放每次出队列的TreeNode
        List<Integer> list = new ArrayList<Integer>();
        //循环控制每次队列的次数,实现将每层元素分离
        for(int i = 0; i < count; i++){
        //出队列
        TreeNode node = queue.poll();
        list.add(node.val);
        //将poll出的节点的左右节点(下一层入队列),此时丢列先进先出,后进后出的特点实现了层序遍历(也是实现了BFS)
        if(node.left != null){
            queue.add(node.left);
        }
        if(node.right != null){

             queue.add(node.right);
        }
    }
    //将这次得到的层序遍历结果存入res中
        res.add(list);

    }
    return res;
}
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值