从上到下打印二叉树II

从上到下打印二叉树II

/*	剑指 Offer32 :从上到下打印二叉树 II
 * 	题目描述：从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。
 * 	解题思路：相较于普通的层序遍历，本题加了每一层打印一行的限制
 * 			 			3
 * 					9		20
 * 						15		7
 * 			  两个问题：1.如何确定树有多深 2.如何分层
 * 			  返回值要求链表啊，不用事先确定有几层，每层有几个
 * 			  这就比较简单了，根结点入队、根结点出队，根节点的所有孩子入队，
 *           队中所有元素出队，他们的孩子再入队
 * */
public class CongShangDaoXiaDaYinErChaShu_II {
	public static void main(String[] args) {
		TreeNode root = new TreeNode(3);
		root.left=new TreeNode(9);
		root.right=new TreeNode(20);
		root.right.left=new TreeNode(15);
		root.right.right=new TreeNode(7);
		CongShangDaoXiaDaYinErChaShu_IISolution solution = new CongShangDaoXiaDaYinErChaShu_IISolution();
		List<List<Integer>> result = solution.levelOrder(root);
		Iterator<List<Integer>> iterator1 = result.iterator();
		while(iterator1.hasNext()) {
			List<Integer> list = iterator1.next();
			Iterator<Integer> iterator2 = list.iterator();
			System.out.println("[");
			while(iterator2.hasNext()) {
				System.out.print(iterator2.next()+" ");
			}
			System.out.println("]");
		}
	}
}

class CongShangDaoXiaDaYinErChaShu_IISolution {
    public List<List<Integer>> levelOrder(TreeNode root) {
    	if(root==null) return new LinkedList<List<Integer>>();
    	List<List<Integer>> list = new LinkedList<List<Integer>>();
    	Queue<TreeNode> queue = new LinkedList<TreeNode>();
    	queue.add(root);
    	while(!queue.isEmpty()) {
    		//将队列中的所有元素出队倒腾进另一个队列
    		Queue<TreeNode> tempQueue = new LinkedList<TreeNode>();
    		List<Integer> tempList = new LinkedList<Integer>();
    		while(!queue.isEmpty()) {
    			tempQueue.add(queue.poll());
    		}
    		//将暂存队列中的全部元素出队并接入结果集，他们的孩子入队
    		while(!tempQueue.isEmpty()) {
    			TreeNode node = tempQueue.poll();
    			tempList.add(node.val);
    			if(node.left!=null) queue.add(node.left);
    			if(node.right!=null) queue.add(node.right);
    		}
    		list.add(tempList);
    	}
    	return list;
    }
}