从上到下打印二叉树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;
}
}