LeetCode每日一题——1161. 最大层内元素和

MaTF_

已于 2023-12-14 15:36:32 修改

阅读量150

点赞数 1

分类专栏： LeetCode每日一题文章标签： leetcode 深度优先算法

于 2022-07-31 10:47:16 首次发布

本文链接：https://blog.csdn.net/MaTF_/article/details/126082843

版权

LeetCode每日一题专栏收录该内容

6 篇文章 0 订阅

订阅专栏

题目传送门

题目描述

在这里插入图片描述

示例

在这里插入图片描述

思路

本题是对dfs的简单应用。

代码

/**
 * 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 level_sum[10005];
    int max_level = 0;
public:
    void dfs(TreeNode* x, int d){
        max_level = max(max_level, d);
        level_sum[d] += x->val;
        if(x->left!=nullptr) dfs(x->left, d+1);
        if(x->right!=nullptr) dfs(x->right, d+1);
    }
    int maxLevelSum(TreeNode* root) {
        dfs(root, 1);
        int sum=1<<31, ans=0;
        for(int i=1;i<=max_level;i++){
            if(level_sum[i]>sum){
                sum = level_sum[i];
                ans = i;
            }
        }
        return ans;
    }
};