标签:树的直径 htm mes 函数 algo 更新 节点 etc 之间
https://leetcode-cn.com/problems/binary-tree-maximum-path-sum/
// Problem: LeetCode 124
// URL: https://leetcode-cn.com/problems/binary-tree-maximum-path-sum/
// Tags: Tree Recursion DFS
// Difficulty: Hard
#include <iostream>
#include <algorithm>
using namespace std;
struct TreeNode
{
int val;
TreeNode *left;
TreeNode *right;
};
class Solution
{
private:
int maxSum = INT_MIN; // 全局最大路径和
// 功能:求一颗树的最大路径和,要求从根节点开始,在左子树或右子树结束,也可不经过左子树和右子树,即至多经过一个子树
int dfs(TreeNode *root)
{
if (root == nullptr)
return 0;
int left = dfs(root->left);
int right = dfs(root->right);
// 过程中更新全局最大路径和:根节点必须经过,左右子树可经过可不经过
maxSum = max(maxSum, root->val + max(left, 0) + max(right, 0));
// 从根节点开始,在左子树或右子树结束,也可不经过左子树和右子树,即至多经过一个子树
return max(root->val, root->val + max(max(left, right), 0));
}
public:
// 求一颗树的最大路径和,不要求经过根节点,起始和终止节点任意
int maxPathSum(TreeNode *root)
{
dfs(root);
return this->maxSum;
}
};
作者:@臭咸鱼
转载请注明出处:https://www.cnblogs.com/chouxianyu/
欢迎讨论和交流!
标签:树的直径 htm mes 函数 algo 更新 节点 etc 之间
原文地址:https://www.cnblogs.com/chouxianyu/p/13386394.html