题目:
二叉树的题目要先从问题的角度分析微观结构,即分析清楚二叉小树(如图1),然后适配大树。
图1
按照题意:分析图1的小树的所有情况:
1.根 + 左 + 右;
2.根 + 左;
3.根 + 右;
4.根;
5.左;
6.右;
站在小树(微观)的角度考虑问题:分析上述6种情况, 情况2,3,4 可以以链状的形式进一步向a节点的父节点传递,而情况1,5,6则没法向a的父节点传递(情况1向上传递的话会出现分叉,5,6显然),因此需要一个全局变量存储 1,5,6这三种不可传递的结果。
站在大树(宏观)的角度考虑问题:每个节点都应该对外返回【一端连接本节点的最大路径和】,同时保存那些无法向上传递组成更长路径的结果,然后递归调用。
C++实现如下
/**
* 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 maxPathSum(TreeNode* root) {
(void)subLine(root);
return ans;
}
private:
int ans = INT_MIN; // 所有其他结果的最大值
// func:返回一个端点(上端)是root节点的一条路径和(即:左或右半边的一条路径+root节点)
int subLine(TreeNode* root) {
if (root == nullptr) {
return 0;
}
// 若返回为负,则不接入
int left = max(0, subLine(root->left));
int right = max(0, subLine(root->right));
// 保存不应该向上传递的结果
ans = max(ans, left + right + root->val);
// 返回子路径和(含root),用于进一步向上集中
return max(left, right) + root->val;
}
};