携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第18天,点击查看活动详情
题目描述
给你一棵二叉树的根节点 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
提示:
- 树中节点数目在范围 [1, 104] 之间。
- 1 <= Node.val <= 100
思路
本题要求出层数最深的叶子节点的和,其实就是求最下一层节点的和,因为既然是最下面一层的节点,就一定是叶子节点了。我们可以安全层序遍历的方式,求出每1层的和,那么最后1次求出的和就是层数最深的叶子节点的和。
层序遍历的方法:
- 定义1个队列
- 将根节点加入队列
- 如果队列非空,进入while循环
- 获取队列的大小,依次取出
- 取出节点如果存在左子节点或者右子节点,将子节点加入队列中
不难发现,通过这样遍历,每次遍历的节点都是在同一层的,第5步讲子节点加入队列是给下一次循环遍历使用的,本次循环遍历通过第4步控制了大小,不会遍历到下一层的节点。
本题就是在循环遍历的时候,对每一层进行求和,这样非常容易理解,但是缺陷在于对于求和了很多次,特别是深度较深的时候,前面的几次求和都是有浪费的。如果要避免这种情况,需要先遍历去到最深的层数,但是这样就要遍历2次,只是一个取舍。
Java版本代码
class Solution {
public int deepestLeavesSum(TreeNode root) {
Queue<TreeNode> queue = new ArrayDeque<>();
queue.add(root);
int sum = 0;
while (!queue.isEmpty()) {
sum = 0;
int size = queue.size();
for (int i = 0; i < size; i++) {
TreeNode node = queue.poll();
sum += node.val;
if (node.left != null) {
queue.add(node.left);
}
if (node.right != null) {
queue.add(node.right);
}
}
}
return sum;
}
}