leetcode || 124、Binary Tree Maximum Path Sum

最新推荐文章于 2024-09-24 08:39:28 发布

hustyangju

最新推荐文章于 2024-09-24 08:39:28 发布

阅读量684

点赞数

分类专栏： LeetCode 文章标签： leetcode 二叉树路径和 DFS

本文链接：https://blog.csdn.net/hustyangju/article/details/45307843

版权

LeetCode 专栏收录该内容

131 篇文章 1 订阅

订阅专栏

problem：

Given a binary tree, find the maximum path sum.

The path may start and end at any node in the tree.

For example:
Given the below binary tree,

       1
      / \
     2   3

Return 6.

Hide Tags

Tree Depth-first Search

题意：在一棵二叉树中寻找一条路径，使其和最大

thinking：

（1）二叉树寻找一条路径比较难做，没有parent指针更难

（2）采用DFS遍历，从根结点开始

code：

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int calLen(TreeNode *root, int &len)
    {
        if (root == NULL)
        {
            len = 0;
            return 0;
        }
        
        if (root->left == NULL && root->right == NULL)
        {
            len = root->val;
            return root->val;
        }
        
        int leftPath, rightPath;
        int leftLen;
        if (root->left)
            leftLen = calLen(root->left, leftPath);
        else
        {
            leftLen = INT_MIN;
            leftPath = 0;
        }
        
        int rightLen;
        if (root->right)
            rightLen = calLen(root->right, rightPath);
        else
        {
            rightLen = INT_MIN;
            rightPath = 0;
        }
        
        len = max(max(leftPath, rightPath) + root->val, root->val);
        int maxLen = max(root->val, max(leftPath + rightPath + root->val, 
            max(leftPath + root->val, rightPath + root->val)));
        
        return max(max(leftLen, rightLen), maxLen);
    }
    
    int maxPathSum(TreeNode *root) {
        int len;
        return calLen(root, len);
    }
};