题目链接
思路:
在二叉树层次遍历的基础上进行修改,每次弹出这一整层的节点,得到每一层的所有节点
先将根节点加入队列中,进入循环时,每一层的节点个数等于当前队列的大小,利用此性质来得到当前层的所有节点
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> ans = new ArrayList<>();
if (root == null) {
return ans;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
while (!queue.isEmpty()) {
// 当前层的大小,即节点个数,等于当前队列的大小
int size = queue.size();
List<Integer> curLevel = new ArrayList<>();
while (size > 0) {
TreeNode cur = queue.poll();
curLevel.add(cur.val);
// 有左孩子
if (cur.left != null) {
queue.add(cur.left);
}
// 有右孩子
if (cur.right != null) {
queue.add(cur.right);
}
size--;
}
// 当前层结果加入最终结果
ans.add(curLevel);
}
return ans;
}
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}