二叉树中的最大路径和

最新推荐文章于 2022-01-13 06:45:05 发布

AquilaEAG

最新推荐文章于 2022-01-13 06:45:05 发布

阅读量709

点赞数

分类专栏：算法题

本文链接：https://blog.csdn.net/weixin_43202635/article/details/108021289

版权

算法题专栏收录该内容

71 篇文章 1 订阅

订阅专栏

二叉树中的最大路径和

题目

给定一个非空二叉树，返回其最大路径和。

本题中，路径被定义为一条从树中任意节点出发，达到任意节点的序列。该路径至少包含一个节点，且不一定经过根节点。

示例 1:

输入: [1,2,3]
1
/
2 3
输出: 6

示例 2:

输入: [-10,9,20,null,null,15,7]
-10
/
9 20
/
15 7
输出: 42

思路

在这里插入图片描述

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
private:
    int ans;
public:
    int maxPathSum(TreeNode* root) {
        ans = INT32_MIN;
        dfs(root);
        return ans;
    }

    int dfs(TreeNode* root) {
        if (!root) return 0;

        int leftMaxPathSum = max(dfs(root->left), 0);
        int rightMaxPathSum = max(dfs(root->right), 0);
        ans = max(ans, root->val+leftMaxPathSum+rightMaxPathSum);

        return root->val+max(leftMaxPathSum, rightMaxPathSum);
    }
};