leetcode:124. 二叉树中的最大路径和

题目描述:

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

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

示例 1:

输入: [1,2,3]

       1
      / 
     2   3

输出: 6

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

   -10
   / 
  9  20
    /  
   15   7

输出: 42

思路分析:

路径问题常规想到的就是用搜索解决。这道题用到了dfs,用递归完成。对于每个结点,计算其左右子树的贡献值,更新当前的最大路径为原始最大路径和左子树贡献加当前结点加右子树贡献和中较大的一个。需要注意的是计算左右子树贡献的时候,需要将取贡献值和0中的较大值,因为结点权值可能为负。同时递归函数的返回值应该为左右子树中贡献较大的加上当前结点权值。

时间复杂度为O(n),n为树中结点个数。

代码:

 1 /**
 2  * Definition for a binary tree node.
 3  * struct TreeNode {
 4  *     int val;
 5  *     TreeNode *left;
 6  *     TreeNode *right;
 7  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 8  * };
 9  */
10 class Solution {
11 public:
12     int max_sum = INT_MIN;
13     int get_max(TreeNode* cur)
14     {
15         if(cur == NULL)
16             return 0;
17         int leftmax = max(get_max(cur->left), 0);
18         int rightmax = max(get_max(cur->right), 0);
19         max_sum = max(max_sum, leftmax+cur->val+rightmax);
20         return max(leftmax, rightmax) + cur->val;
21     }
22     int maxPathSum(TreeNode* root) {
23         if(!root)
24             return 0;
25         get_max(root);
26         return max_sum;
27     }
28 };
原文地址:https://www.cnblogs.com/LJ-LJ/p/11203368.html