題目要求:https://leetcode.com/problems/binary-tree-level-order-traversal/
1)此題本質上是一個BFS問題,不難解決
- 題目要求將每一層節點作爲一個list,那麼使用while(!q.isEmpty())的方式來遍歷Queue就不太合適,因爲需要不停地將節點加入隊列。
- 這裏在訪問每一層時,先獲得此時queue裏的長度,用for循環遍歷。
- 使用Iterator()同樣可行
AC代碼:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
Queue<TreeNode> q = new LinkedList<TreeNode>();
List<List<Integer>> resList = new LinkedList<List<Integer>>();
if(root == null) return resList;
q.add(root);
while(!q.isEmpty()){
List<Integer> levelList = new LinkedList<Integer>();
int qSize = q.size();
for(int i=0; i<qSize; i++){
TreeNode tmp = q.poll();
levelList.add(tmp.val);
if(tmp.left != null){
q.add(tmp.left);
}
if(tmp.right != null){
q.add(tmp.right);
}
}
resList.add(levelList);
}
return resList;
}
}