
思路:
每次把下一层的数据从左到右放进去,然后取的时候容器里的就是当前层的数据,先把大小拿到,然后取数据同时将非空左右孩子放进去,直到取完该层的数据
代码:
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
if (root == null) {
return res;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while (!queue.isEmpty()) {
List<Integer> temp = new ArrayList<>();
int levelSize = queue.size();
//每次将当前层的结点从左向右取出,将下一层的结点从左向右放进去
for (int i = 0; i < levelSize; i++) {
TreeNode curr = queue.poll();
temp.add(curr.val);
if (curr.left != null) {
queue.offer(curr.left);
}
if (curr.right != null) {
queue.offer(curr.right);
}
}
res.add(temp);
}
return res;
}
}
网友评论