题目要求: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;
}
}