class Solution {
public List<List<Integer>> resList = new ArrayList<List<Integer>>();
public List<List<Integer>> levelOrder(TreeNode root) {
method(root);
return resList;
}
public void method(TreeNode node){
if (node == null) return;
Queue<TreeNode> que = new LinkedList<TreeNode>();
que.offer(node);
while (!que.isEmpty()) {
List<Integer> itemList = new ArrayList<Integer>();
int len = que.size();
while (len > 0) {
TreeNode tmpNode = que.poll();
itemList.add(tmpNode.val);
if (tmpNode.left != null) que.offer(tmpNode.left);
if (tmpNode.right != null) que.offer(tmpNode.right);
len--;
}
resList.add(itemList);
}
}
}
输入:二叉树的根节点,输出:一个列表型列表(即列表中包含子列表)
关于子函数method:
输入:二叉树的根节点
首先判断特殊情况,根节点是否存在;创建队列,将子节点加入队列中
通过队列是否为空作为函数终止条件,设置辅助变量len作为每一层二叉树的长度即子列表的长度,通过len控制子列表的生成
关于子列表的生成,首先将队列中的节点弹出,然后再将其左右子节点放入队列;如果二叉树当前层的节点全部弹出则退出循环,重新更新len的值。
也就是说,队列在弹出当前层的节点时,会将下一层节点添加进队列,当前层的节点全部弹出时,下一层的节点已经全部加入到队列中,周而复始直至二叉树遍历结束,也就是队列中不再有元素。