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]
]
按level输出二叉树的元素,每一层保存在一个list
102思路:
BFS
但是需要两个queue,一个当前level,一个下一level
每遍历完一个queue,交换当前和下一leve的queue
//1ms
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
if(root == null) return res;
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while(!queue.isEmpty()) {
int size = queue.size();
List<Integer> tmp = new ArrayList<>();
for(int i = 0; i < size; i++) {
TreeNode cur = queue.poll();
tmp.add(cur.val);
if(cur.left != null) queue.offer(cur.left);
if(cur.right != null) queue.offer(cur.right);
}
res.add(tmp);
}
return res;
}
104思路:
和102一样BFS
不同之处在于保存到结果list< list>时,不是在末尾插入list,而是在开头插入list
//1ms
public List<List<Integer>> levelOrderBottom(TreeNode root) {
List<List<Integer>> result = new ArrayList<>();
if (root == null) {
return result;
}
List<TreeNode> current = new ArrayList<>();
List<TreeNode> next = new ArrayList<>();
current.add(root);
while (!current.isEmpty()) {
List<Integer> tmp = new ArrayList<>();
for (TreeNode node : current) {
tmp.add(node.val);
if (node.left != null) {
next.add(node.left);
}
if (node.right != null) {
next.add(node.right);
}
}
current.clear();
result.add(0, tmp);
List<TreeNode> tmpList = current;
current = next;
next = tmpList;
}
return result;
}