牛客刷题——求二叉树的层次遍历

题目:

给定一个二叉树,返回该二叉树层序遍历的结果,(从左到右,一层一层地遍历)
例如:
给定的二叉树是{3,9,20,#,#,15,7},

该二叉树层序遍历的结果是
[
[3],
[9,20],
[15,7]
]

 对于不分层直接输出的层次遍历,直接采用队列的思想,首先root入队,针对队列的每一个元素,当前元素出队,访问,当前元素的左右孩子入队,直到队列为空,这里需要进行分层输出,所以在进行访问的时候也需要进行分层处理。主要思想是,每一层采用一个队列,在对当前层遍历的同时,将当前层的孩子放入到新的队列,对当前层遍历结束后,对新的队列进行遍历。

完整代码如下:

 public  ArrayList<ArrayList<Integer>> levelOrder (TreeNode root) {
        // write code here
		
		List<TreeNode> nodes = new ArrayList<>();
		nodes.add(root);
		ArrayList<ArrayList<Integer>> res = new ArrayList<>(); 
		int count = 0;
		while(nodes.size()!=0) {
			List<TreeNode> temp = new ArrayList<>();   //构建新的队列
			ArrayList<Integer> list = new ArrayList<>();   //当前层的遍历结果
			for(int i=0;i<nodes.size();i++) {
				TreeNode now = nodes.get(i);
				if(now!=null) {
					list.add(now.val);
					temp.add(now.left);
					temp.add(now.right);
				}		
			}
			nodes=temp;
            if(list.size()>0){
                res.add(list);

            }
		}
		return res;
    }

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值