从上到下打印二叉树
从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
核心思想:要求每层单独打印,可以根据当前队列中元素的个数判断需要循环几次
for(int i = queue.size(); i > 0; i--)
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> list = new ArrayList<List<Integer>>();
Queue<TreeNode> queue = new LinkedList<TreeNode>();
if(root != null){
queue.add(root);
}
while(!queue.isEmpty()){
List<Integer> temp = new ArrayList<Integer>();
//关键点
for(int i = queue.size(); i > 0; i--){
TreeNode cur = queue.poll();
temp.add(cur.val);
if(cur.left != null){
queue.add(cur.left);
}
if(cur.right != null){
queue.add(cur.right);
}
}
list.add(temp);
}
return list;
}
}