給定一個二叉樹,返回其按層次遍歷的節點值。 (即逐層地,從左到右訪問所有節點)。
例如:
給定二叉樹:[3,9,20,null,null,15,7]
,3 / \ 9 20 / \ 15 7返回其層次遍歷結果:
[ [3], [9,20], [15,7] ]
我的解答:
/**
* 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>> lists = new ArrayList<List<Integer>>();
// 不能return null;不然跑不過測試用例
if (root == null) {
return lists;
}
// 存儲每層的node
List<TreeNode> nodes = new ArrayList<TreeNode>();
nodes.add(root);
boolean flag = true;
while (flag) {
List<TreeNode> children = new ArrayList<TreeNode>();
List<Integer> list = new ArrayList<Integer>();
for (TreeNode i : nodes) {
list.add(i.val);
if (i.left != null) {
children.add(i.left);
}
if (i.right != null) {
children.add(i.right);
}
}
lists.add(list);
// 替換nodes
nodes = children;
if (nodes.isEmpty()) {
flag = false;
}
}
return lists;
}
}