LeetCode102：二叉树的层次遍历

版权声明：本文为博主原创文章，转载请注明出处！ https://blog.csdn.net/ASN_forever/article/details/84869190

 

思路：用一个队列和列表来实现。主要是两层循环，外层循环用来获取队列大小size以及初始化一个列表list（用来存储当前层的节点），内层循环用来依次从队列中从左到右的取出当前层节点放到list中，并将当前节点的左右子结点插入到队尾。每次从队列中取出一个节点的时候都要对size进行自减操作，直到当size==0时，也就是当前层所有的节点都从队列中取出后，再重新开始外层循环，计算下一层的结点数size，并初始化一个用来保存当前层节点的列表list，进行新一轮的循环，以此类推。直到最后队列为空的时候，说明当前层没有节点了，算法结束。

public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<List<Integer>>();
        if(root==null)
            return res;
        Queue<TreeNode> q = new ArrayDeque<TreeNode>();
		TreeNode left = null;
		TreeNode right = null;
		TreeNode temp = root;
		q.add(root);
		int size = 1;
		while(true){
			size = q.size();
			ArrayList<Integer> inner = new ArrayList<Integer>();
			while(size!=0){
				temp = q.poll();
				inner.add(temp.val);
				size--;
				left = temp.left;
				right = temp.right;
				if(left!=null)
					q.add(left);
				if(right!=null)
					q.add(right);
			}
			res.add(inner);
			if(q.size()==0)
				break;
		}
		
		return res;
    }