携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第30天,点击查看活动详情
题目描述
给你一棵二叉树的根节点 root ,请你返回 层数最深的叶子节点的和 。
示例 1:
输入:root = [1,2,3,4,5,null,6,7,null,null,null,null,8]
输出:15
示例 2:
输入:root = [6,7,8,2,7,1,3,9,null,1,4,null,null,null,5]
输出:19
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/deepest-leaves-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路分析
- 今天的算法题目是二叉树的题目,题目要求返回 层数最深的叶子节点的和。
- 分析题目之后,我们分为两步求解,第一步,确定当前的层的累加和,第二步,确定当前层是最深的层。如何求当前层的累加和呢?我们一般使用BFS。BFS 全称是 Breadth First Search,中文名是宽度优先搜索,也叫广度优先搜索,是图上最基础、最重要的搜索算法之一。所谓宽度优先。就是每次都尝试访问同一层的节 点。 如果同一层都访问完了,再访问下一层。
- 在Java中,我们使用队列Queue来存储元素,完成BFS相关的功能,首先把 root 放入队列,然后每次计算出当前队列的长度,就是当前层元素的个数,逐个出队,完成当前层元素求和计算。
- 在使用BFS的过程中,已经完成了分层求和,最后一层的和就是答案。 具体实现代码如下,请参考。
通过代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public int deepestLeavesSum(TreeNode root) {
int ans = 0;
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while (!queue.isEmpty()) {
ans = 0;
int size = queue.size();
for (int i = 0; i < size; i++) {
TreeNode node = queue.poll();
ans += node.val;
if (node.left != null) {
queue.offer(node.left);
}
if (node.right != null) {
queue.offer(node.right);
}
}
}
return ans;
}
}
总结
- 上述算法的时间复杂度是O(n), 空间复杂度是O(n)
- 坚持算法每日一题,加油!