515. 在每个树行中找最大值
给定一棵二叉树的根节点 root
,请找出该二叉树中每一层的最大值。
示例1:
输入: root = [1,3,2,5,3,null,9]
输出: [1,3,9]
示例2:
输入: root = [1,2,3]
输出: [1,3]
提示:
- 二叉树的节点个数的范围是 [ 0 , 1 0 4 ] [0,10^4] [0,104]
- − 2 31 ≤ N o d e . v a l ≤ 2 31 − 1 -2^{31} \leq Node.val \leq 2^{31} - 1 −231≤Node.val≤231−1
解法一(BFS+队列)
思路分析:
- 使用二叉树的层序遍历,对每层进行遍历
- 在对每层进行遍历时,寻找每层的最大值,并保存
实现代码如下:
class Solution {
public List<Integer> largestValues(TreeNode root) {
List<Integer> ans = new ArrayList<>();
if (root == null)
return ans; // 排除边界条件
Queue<TreeNode> queue = new ArrayDeque<>();
queue.offer(root);
while (!queue.isEmpty()) {
int size = queue.size();
int max = Integer.MIN_VALUE; // 标记该层最大值
for (int i = 0; i < size; ++ i) {
TreeNode node = queue.poll();
max = Math.max(max, node.val);
if (node.left != null) queue.offer(node.left);
if (node.right != null) queue.offer(node.right);
}
ans.add(max); // 记录保存每层最大值
}
return ans;
}
return ans;
}
}
提交结果如下:
解答成功:
执行耗时:2 ms,击败了87.47% 的Java用户
内存消耗:44.1 MB,击败了9.26% 的Java用户
复杂度分析:
- 时间复杂度: O ( n ) O(n) O(n),遍历二叉树
- 空间复杂度: O ( n ) O(n) O(n),辅助队列
解法二(递归+dfs)
思路分析:
- 可以使用递归深度优先搜索去寻找每层的最大值。
- 首先思考递归的参数和返回值,因为需要寻找每层的最大值,所以需要一个变量来标记节点所在层数,同时需要传递二叉树的节点,以及需要保存结果的列表参数;同时不需要返回值
- 思考递归的边界条件,对于递归寻找某层二叉树结点的最大值,当结点为空时,即不需要判断是否为最大值,直接返回即可,且此时也意味着遍历到二叉树末端
- 思考递归的过程,对于某个二叉树结点,首先需要判断其所在层 在此之前是否有递归到过,如果有,则将其结点值与此时列表中的结点值进行比较,若没有递归过,则直接将其保存到结果列表中
实现代码如下:
class Solution {
// 递归+dfs
public List<Integer> largestValues(TreeNode root) {
List<Integer> ans = new ArrayList<>();
dfs(root, 0, ans);
return ans;
}
private void dfs(TreeNode node, int deeply, List<Integer> ans) {
if (node == null)
return ; // 递归边界
if (deeply >= ans.size()) { // 该层未曾出现过
ans.add(node.val);
} else { // 若该层出现过 比较得出更大值
Integer num = ans.get(deeply);
if (num < node.val) {
ans.set(deeply, node.val);
}
}
dfs(node.left, deeply+1, ans);
dfs(node.right, deeply+1, ans);
}
}
提交结果如下:
解答成功:
执行耗时:1 ms,击败了96.80% 的Java用户
内存消耗:44.4 MB,击败了5.04% 的Java用户
复杂度分析:
- 时间复杂度: O ( n ) O(n) O(n)
- 空间复杂度: O ( n ) O(n) O(n)