劍指 Offer 32 - II. 從上到下列印二叉樹 II
從上到下按層列印二叉樹，同一層的節點按從左到右的順序列印，每一層列印到一行。
例如:
給定二叉樹: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其層次遍歷結果：
[
[3],
[9,20],
[15,7]
]
提示：
節點總數 <= 1000
一、遍歷（BFS）
本題首先要設定一個佇列queue，一個list，一個臨時列表。
演算法流程大概如下：
首先把頭結點放入queue，然後再把頭結點放入tmp，最後放入list。
然後把左/右節點依次放入佇列，再放入tmp，最後再放入list。
大概的思路，就是tmp作為輔助列表，作為佇列queue和list的橋樑。
然後什麼叫BFS，得解釋一下，是二叉數的廣度優先搜尋的意思。
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
Deque<TreeNode> queue = new LinkedList<>();
List<List<Integer>> res = new ArrayList<>();
//如果頭節點不為空，則放入佇列queue
if (root != null) queue.add(root);
while (!queue.isEmpty()) {
//設定一個輔助陣列，專門裝佇列queue的值
List<Integer> tmp = new ArrayList<>();
for (int i = queue.size(); i > 0; i--) {
TreeNode node = queue.poll();
tmp.add(node.val);
if (node.left != null) queue.add(node.left);
if (node.right != null) queue.add(node.right);
}
res.add(tmp);
}
return res;
}
}
二、遞迴
class Solution {
List<List<Integer>> res;
public List<List<Integer>> levelOrder(TreeNode root) {
res = new ArrayList<>();
dfs(0, root);
return res;
}
void dfs(int depth,TreeNode root) {
if (root == null) return;
if (depth == res.size()) {//如果陣列list的size等於數組裡面depth的話，則建立新的一層節點
res.add(new ArrayList<>());
}
//因為dfs(0,root) 所以這個時候是res.get(0).add(root.val)，是放入這個陣列中第一個陣列的頭節點
res.get(depth).add(root.val);
//遞迴遍歷左右節點
dfs(depth + 1, root.left);
dfs(depth + 1, root.right);
}
}