剑指offer 78:把二叉树打印成多行

把二叉树打印成多行

从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
在这里插入图片描述
即每一层的所有节点数作为一个单独的List !

思路:
本题是BFS广度优先的变形。 考点就在于按照节点个数来分层 !
结论: 每一层在弹出队列nodes中 本层的最后一个节点时,此时队列nodes中装的正好就是下一层的所有节点!
即每一层的节点个数为 nodes.size()
所以使用for循环,将nodes的size() 作为下一层去遍历的次数 ! 即可以达到分层!

注意: 在for中弹出nodes!

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        // 把每一层当成一个数组 !
        List<List<Integer>> res=new ArrayList<>();
        Queue<TreeNode> nodes=new LinkedList<>();
         if(root==null){  //安全校验
            return res;  
        }
        
        nodes.add(root); // 先添加头节点
        while(!nodes.isEmpty()){
            List<Integer> list=new ArrayList<>(); //  每一层一个list
            int size=nodes.size();   //先记录每一层的节点数,如果用nodes.size(),则可能会变化而出错
            // 每一个for就是一层,都放进一个list中
            for(int i=0;i<size;i++){  
                TreeNode n=nodes.poll();  
                list.add(n.val);  //将本层的都装入list
                if(n.left!=null){   
                    nodes.add(n.left); //下层的都压入nodes
                }
                if(n.right!=null){
                    nodes.add(n.right);
                }
            }
            //本层遍历结束,将list装入res
            res.add(list);
        }
        return res;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值