32-2 從上到下打印二叉樹 II
面試題32 - II. 從上到下打印二叉樹 II
難度簡單17
從上到下按層打印二叉樹,同一層的節點按從左到右的順序打印,每一層打印到一行。
例如:
給定二叉樹: [3,9,20,null,null,15,7],
3 / \ 9 20 / \ 15 7
返回其層次遍歷結果:
[ [3], [9,20], [15,7] ]
1.遞歸
//遞歸
List<List<Integer>> list = new ArrayList();
public List<List<Integer>> levelOrder(TreeNode root) {
recur(root,0);
return list;
}
public void recur(TreeNode root,int k){
//終止條件
if(root != null){
if(list.size()<=k){
list.add(new ArrayList());
}
list.get(k).add(root.val);
recur(root.left,k+1);
recur(root.right,k+1);
}
}
2.隊列+迭代
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> result = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
//添加根節點
queue.add(root);
while(!queue.isEmpty()){
List<Integer> list = new ArrayList<>();
int cnt = queue.size();
while(cnt-- > 0){
TreeNode t = queue.poll();//彈出首節點
if(t == null){
continue;
}
list.add(t.val);
queue.add(t.left);
queue.add(t.right);
}
if(list.size() != 0){
result.add(list);
}
}
return result;
}