思路:
根据上一道题的思想,我们依旧采取队列的存储,只是将子节点存储到新的队列,把当前队列中的值全部获取存储到一个容器中,再存储到结果集中罢了。
第一步,我们先把根节点放入队列,
第二步,遍历队列,把队列中的节点的值放到一个容器中,用来保存当前这样一层打印的结果,判断这些节点是否有孩子节点,如果有孩子节点,顺序放入一个新的队列中。并且处理完一个节点就让一个节点出队。
第三步,判断这一行的结果是否为Empty,如果为Empty,那么就不需要加入到结果集中。判断是否有下一行节点,也就是新的队列的节点是否为Empty,如果没有新的节点,那么就证明已经是最后一层了;如果有新的节点,那么新的队列重复第二步和第三步。
由此看来我们可以用递归去实现。
public List<List<Integer>> levelOrder(TreeNode root) {
if(root == null) return null;
LinkedList<TreeNode> queue = new LinkedList();
//首先用queue保存序列
queue.add(root);
List<List<Integer>> result = new ArrayList();
hasChild(result ,queue);
return result;
}
public void hasChild(List result, LinkedList<TreeNode> queue){
//保存下一层的节点
ArrayList<Integer> newLine = new ArrayList();
//保存当前队列中节点的孩子节点。
LinkedList<TreeNode> newQueue = new LinkedList();
while(!queue.isEmpty()){
TreeNode currentNode = queue.removeFirst();
newLine.add(currentNode.val);
if(currentNode.left != null) newQueue.add(currentNode.left);
if(currentNode.right != null) newQueue.add(currentNode.right);
}
if(!newLine.isEmpty()){
result.add(newLine);
}
else{
return;
}
if(!newQueue.isEmpty()){
hasChild(result , newQueue);
}
}