剑指offer32-II.从上到下打印二叉树

题目: 从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一个list里。

题目分析

题目的本质是层序遍历,

题解思路

思路: 初始化一个队列,非空的root节点入队列,根据此时队列的长度,依次出队列;出队列的同时将该节点的非空子节点入队列,循环,直到队列为空。
代码

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<>();
        List<List<Integer>> res = new ArrayList<>();
        if(root != null) queue.add(root);
        while(!queue.isEmpty()) {
            List<Integer> tmp = new ArrayList<>();
            for(int i = queue.size(); i > 0; i--) {
                TreeNode node = queue.poll();
                tmp.add(node.val);
                if(node.left != null) queue.add(node.left);
                if(node.right != null) queue.add(node.right);
            }
            res.add(tmp);
        }
        return res;
    }
}

与我的方法的不同之处就在于循环控制条件,for(int i = queue.size(); i > 0; i--),循环控制的i初始化为queue.size,每次循环-1即使循环中往队列里加入了新的节点(即下一层的节点),也不会影响上一层节点的输出。


记录我的思路…道阻且长啊

迭代

思路: 初始化一个队列,按照层数将节点依次入队列,随着该层节点依次出队列并将val值存入List中,将这些节点的左右子节点分别入队列,重复此过程,直到队列为空。

  • 若是满二叉树的情况下,二叉树的每层的树节点为 2 n − 1 2^{n-1} 2n1 (n为层数),因此可用此数作为循环控制的条件,但是这里的二叉树并不一定是满二叉树,入队列时空节点该如何处理?——对于这个问题,其实类似问题在剑指offer28题中出现过,使用LinkedList而非ArrayDeque可以解决这个问题。就是非要把空节点考虑进去…
  • 在写完代码运行的时候发现有些测试用例通不过,各种调试之后各种针对bug修补之后,形成了以下代码:
public List<List<Integer>> levelOrder(TreeNode root){
		Queue<TreeNode> queue = new LinkedList<TreeNode>();
		List<List<Integer>> list_all = new LinkedList<List<Integer>>();
		if(root==null) {
			return list_all;
		}		
		int nodeNum = 1;
		queue.add(root);
		while(!queue.isEmpty()) {
			List<Integer> list_layer = new LinkedList<Integer>();
			int flag=0;
			for(int i=0;i<nodeNum;i++) {
				TreeNode node = queue.poll();
				if(node!=null) {
					list_layer.add(node.val);				
					if(node.left!=null) {
						queue.add(node.left);
					}else flag++;
					if(node.right!=null) {
						queue.add(node.right);
					}else flag++;					
				}		
			}
			if(list_layer.size()>0) {
				list_all.add(list_layer);
			}
			nodeNum = nodeNum << 1;		
			if(flag!=0) nodeNum-=flag;
		}	
		return list_all;
	}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值