leetcde124二叉树的最大路径和 树型dp

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

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

示例 1:

输入: [1,2,3]

   1
  / \
 2   3

输出: 6

示例 2:

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

   -10
   / \
  9  20
    /  \
   15   7

输出: 42

通过次数48,250
提交次数115,980

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-tree-maximum-path-sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

如图

• 考虑子树x,则可以从4个方向转移过来

  1. 只要x节点
  2. 要x+x->left
  3. 要’x+x->right
  4. 要x+x->left+x->right

• 转移方程: $dp[x]=max(1,2,3,4)$

• 可以推广到多叉树:
  定义vx为根的值,vchl[i]为子树i返回的值,如果vchl[i]是负数就返回0
  $dp[x]=max(vx,dp[chl]+vx)$

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */

typedef TreeNode Node;
#define lef left
#define rig right

class Solution {
public:
    int ans, tmax;
    int dfs(Node* u) {
        if(!u) return -0x3f3f3f;
        tmax = max(u->val, tmax);
        if(!u->lef && !u->rig) {
            ans = max(ans, u->val);
            return max(u->val, 0);
        }
        int ret = u->val;
        int l = dfs(u->lef);
        int r = dfs(u->rig);
        ret = max(ret+l, ret+r);
        ans = max(ans, ret);
        ans = max(ans, l+r+u->val);
        //4个方向里取最大值
        // printf("%d : %d\n", u->val, ans);
        return max(ret, 0);
    }

    int maxPathSum(TreeNode* root) {
        int INF = 0x3f3f3f3f;
        tmax = ans = -INF;
        int p = dfs(root);
        ans = max(ans, p);
        return tmax > 0 ? ans : tmax;
    }
};