LeetCode每日一题打卡（6） 1302. 层数最深叶子节点的和

2022.8.17 每日一题

1302. 层数最深叶子节点的和

题目描述

给你一棵二叉树的根节点 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，感觉没有什么难度，直接模拟然后最简单的深度优先搜索，代码也很好理解。