1. 题目描述
Given a binary tree, return the level order traversal of its nodes’ values. (ie, from left to right, level by level).
For example:
Given binary tree [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
return its level order traversal as:
[
[3],
[9,20],
[15,7]
]
题目的意思为,给定一个二叉树,返回一个二维List,每个list中保存二叉树中对应层的节点。读题要仔细,一开始我就只看图理解成了树的广度优先遍历:(。
2. 解题思路
输的广度优先遍历的变种题,在原使用队列实现的广度优先遍历算法中添加了一个标记,将下一层孩子划分成一段。因为栈里会同时存在当前层节点,和下一层节点,需要有个标记记录他们的分割处。
3. Code
// util包真有用呀
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
if(root == null)
{
// return null;
// 一开始觉得树返回一个空就好了,结果是要返回一个空的List
return new ArrayList<>();
}
List<List<Integer>> result = new ArrayList<>();
Queue<TreeNode> tempQueue = new LinkedList<>();
// offer代替add,在超出队列最大长度时不会抛出异常
tempQueue.offer(root);
ArrayList<Integer> arrayList = new ArrayList<>();
arrayList.add(root.val);
result.add(arrayList);
int num = tempQueue.size();
while(!num.size() == 0)
{
// 新建一个数组,保存下一层的结果
ArrayList<Integer> newList = new ArrayList<>();
while(num > 0) {
// poll代替remove,队列为空时返回null不抛出异常
TreeNode tempNode = tempQueue.poll();
--num;
// 插入当前节点的左右孩子,和其值
if (tempNode.left != null) {
// 队列加入左节点
tempQueue.offer(tempNode.left);
// 结果中加入左节点值
newList.add(tempNode.left.val);
}
if (tempNode.right != null) {
tempQueue.offer(tempNode.right);
newList.add(tempNode.right.val);
}
}
// 和下层孩子节点分割位置
num = tempQueue.size();
if (!newList.isEmpty())result.add(newList);
}
return result;
}
}