Leetcode - Binary Tree Maximum Path Sum

最新推荐文章于 2024-03-25 21:39:45 发布

原创最新推荐文章于 2024-03-25 21:39:45 发布 · 950 阅读

0 ·

CC 4.0 BY-SA版权

Algorithm 专栏收录该内容

78 篇文章

订阅专栏

本文介绍了一种使用递归方法解决树形结构中寻找包含一次转折的最大路径和的问题。通过两个辅助函数，分别计算以当前节点为转折点时的最大路径和以及从当前节点到叶子节点的最大路径和。确保在处理负数路径时，能够正确地跳过无益的分支。

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

解题的关键在于这条路径只能是先往上走，到达某个最高点，再往下走，换句话说，只能有一次转折的机会。所以递归这棵树，记录以某个子节点为转折点时的最大值。值得注意的是树节点的值有负值，所以如果某个子路径的和小于0，放弃它（设置和为0）。

class Solution {
public:
	int maxPathSum(TreeNode *root) {

		int maxSum = -1 << 30;

		int leftMax = pathMaxSum(root->left, maxSum);
		if (leftMax < 0)
			leftMax = 0;
		int rightMax = pathMaxSum(root->right, maxSum);
		if (rightMax < 0)
			rightMax = 0;

		int pathSum = leftMax + rightMax + root->val;
		if (pathSum > maxSum)
			return pathSum;
		else
			return maxSum;
	}


	int pathMaxSum(TreeNode* node, int& maxSum)
	{
		if (node == NULL)
			return 0;

		int leftMax = pathMaxSum(node->left, maxSum);
		if (leftMax < 0)
			leftMax = 0;

		int rightMax = pathMaxSum(node->right, maxSum);
		if (rightMax < 0)
			rightMax = 0;

		if (leftMax + rightMax + node->val > maxSum)//turn down at this point
			maxSum = leftMax + rightMax + node->val;

		int pathMax = leftMax > rightMax ? leftMax : rightMax;
		pathMax += node->val;

		return pathMax;
	}
};