Leetcode Binary Tree Maximum Path Sum

markpen

于 2016-12-22 03:35:25 发布

阅读量326

点赞数

分类专栏： leetcode 文章标签： dfs leetcode

本文链接：https://blog.youkuaiyun.com/markpen/article/details/53807535

版权

leetcode 专栏收录该内容

167 篇文章

订阅专栏

本文介绍了一种求解树中最大路径和的算法实现，采用深度优先搜索(DFS)策略，递归地计算以每个节点为路径一部分的最大值及包含该节点路径的最大和。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：找到一棵树中，各结点值和最大的路径。

思路：DFS，返回必须经过该结点的路径的最大值和可能经过该结点的路径的最大值。

/**
 * 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 {
public:
    int maxPathSum(TreeNode* root) {
        pair<int, int> re;
        re = dfs(root);
        return max(re.first, re.second);
    }
    pair<int, int> dfs(TreeNode* root) {
        pair<int, int> myp;
        if(root == NULL) {
            myp.first = std::numeric_limits<int>::min() / 10000;
            myp.second = std::numeric_limits<int>::min() / 10000;
            return myp;
        }
        
        pair<int, int> leftp;
        pair<int, int> rightp;
        leftp = dfs(root->left);
        rightp = dfs(root->right);
        
        myp.first = max(root->val + max(leftp.first, rightp.first), root->val);
        myp.second = max(max(root->val + leftp.first + rightp.first, root->val), max(leftp.second, rightp.second));
        myp.second = max(myp.first, myp.second);
        
        return myp;
    }
};