0%

leetcode-107

107. 二叉树的层序遍历 II

给定一个二叉树,返回其节点值自底向上的层序遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)

例如:
给定二叉树 [3,9,20,null,null,15,7],

1
2
3
4
5
3
/ \
9 20
/ \
15 7

返回其自底向上的层序遍历为:

1
2
3
4
5
[
[15,7],
[9,20],
[3]
]

方法一:广度优先搜索

树的层次遍历可以使用广度优先搜索实现。从根节点开始搜索,每次遍历同一层的全部节点,使用一个列表存储该层的节点值。

如果要求从上到下输出每一层的节点值,做法是很直观的,在遍历完一层节点之后,将存储该层节点值的列表添加到结果列表的尾部。这道题要求从下到上输出每一层的节点值,只要对上述操作稍作修改即可:在遍历完一层节点之后,将存储该层节点值的列表添加到结果列表的头部。

为了降低在结果列表的头部添加一层节点值的列表的时间复杂度,结果列表可以使用链表的结构,在链表头部添加一层节点值的列表的时间复杂度是 O(1)。在 Java 中,由于我们需要返回的 List 是一个接口,这里可以使用链表实现;而 C++ 或 Python 中,我们需要返回一个 vector 或 list,它不方便在头部插入元素(会

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
public class Solution2 {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> ans = new ArrayList<>();
if (root == null) {
return ans;
}

Queue<TreeNode> deq = new LinkedList<>();
deq.offer(root);
while (!deq.isEmpty()) {
int cnt = deq.size();
List<Integer> level = new ArrayList<>();
for (int i = 0; i < cnt; i++) {
TreeNode node = deq.poll();
level.add(Objects.requireNonNull(node).val);
if (node.left != null) {
deq.offer(node.left);
}
if (node.right != null) {
deq.offer(node.right);
}
}
ans.add(0, level);
}
return ans;
}
}

复杂度分析

时间复杂度:O(n),其中 n 是二叉树中的节点个数。每个节点访问一次,结果列表使用链表的结构时,在结果列表头部添加一层节点值的列表的时间复杂度是 O(1),因此总时间复杂度是 O(n)。

空间复杂度:O(n),其中 n 是二叉树中的节点个数。空间复杂度取决于队列开销,队列中的节点个数不会超过 n。

方法二:深度优先搜索

使用递归算法

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
public class Solution {
private final List<List<Integer>> ans = new ArrayList<>();

public List<List<Integer>> levelOrderBottom(TreeNode root) {
helper(root, 0);
Collections.reverse(ans);
return ans;
}

public void helper(TreeNode root, int level) {
if (root == null) {
return;
}
if (ans.size() == level) {
ans.add(new ArrayList<Integer>());
}
ans.get(level).add(root.val);
helper(root.left, level + 1);
helper(root.right, level + 1);
}
}

复杂度分析

时间复杂度:O(n),其中 n 是二叉树中的节点个数。

空间复杂度:O(n),其中 n 是二叉树中的节点个数。