解析:(使用队列进行层序遍历)
题目要求:给你二叉树的根节点 root ,返回其节点值的 层序遍历 。
大体思路:每次弹出当前层的节点时,将该节点对应的孩子节点加入队列。
每次while弹出当前层所有节点,加上下一层所有节点,所以每一次这样处理完,size()就是下一层所有节点的个数。
如何区分队列中的当前层节点和下一层节点?就是通过上述的size()进行区分
代码:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> list = new ArrayList<>();
LinkedList<TreeNode> queue = new LinkedList<>();
if (root == null) return list;
queue.add(root);
while (!queue.isEmpty()) {
ArrayList<Integer> arr = new ArrayList<>();
// size是当前层的节点个数(由于每次while弹出当前层所有节点,加上下一层所有节点)
int size = queue.size();
while (size > 0) {
TreeNode node = queue.poll();
arr.add(node.val);
if (node.left != null) queue.add(node.left);
if (node.right != null) queue.add(node.right);
size--;
}
list.add(arr);
}
return list;
}
}
给定一个二叉树的根节点,此题讨论如何利用队列进行层次遍历。基本思路是每次从队列中取出当前层的所有节点,然后将这些节点的子节点加入队列。当队列不为空时,队列的大小指示了下一层的节点数。
311

被折叠的 条评论
为什么被折叠?



