2022.8.17 每日一题
题目描述
给你一棵二叉树的根节点 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, ]
之间。 1 <= Node.val <= 100
解法
直接模拟,先深度优先遍历一次树的所有节点,获得最深节点的深度maxd,再同样的方法遍历一次所有节点,当遇到深度等于maxd的节点时答案加上这个节点的值val。两次遍历复杂度都为O(n),所以最终时间复杂度为O(n)。
代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
int ans=0,depth=0,maxd=0;
int deepestLeavesSum(TreeNode* root) {
dfs1(root);
depth=0;
dfs2(root);
return ans;
}
void dfs1(TreeNode* root){
depth++;
if(depth>maxd)
maxd=depth;
if(root->left!=NULL)
{
dfs1(root->left);
depth--;
}
if(root->right!=NULL)
{
dfs1(root->right);
depth--;
}
return;
}
void dfs2(TreeNode* root){
depth++;
if(depth==maxd)
ans+=root->val;
if(root->left!=NULL)
{
dfs2(root->left);
depth--;
}
if(root->right!=NULL)
{
dfs2(root->right);
depth--;
}
return;
}
};
代码思路
ans记录最终答案,depth记录当前深度搜索到的节点深度,maxd记录搜索到的节点最大深度。dfs1为第一次搜索,目的是获得最大节点深度maxd;dfs2为第二次搜索,搜索的过程中累加深度为maxd的节点值,最终得出答案。
总结
给的难度是medium,感觉没有什么难度,直接模拟然后最简单的深度优先搜索,代码也很好理解。