问题描述 :
给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。
示例 1:
输入: [1,2,3]
1
/ \
2 3
输出: 6
示例 2:
输入: [-10,9,20,null,null,15,7]
-10
/ \
9 20
/ \
15 7
输出: 42
可使用以下main函数:
#include <iostream>
#include <queue>
#include <cstdlib>
#include <cstring>
#include<math.h>
using namespace std;
struct TreeNode{
int val;
TreeNode *left;
TreeNode *right;
TreeNode() : val(0), left(NULL), right(NULL) {}
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
};
TreeNode* inputTree(){
int n,count=0;
char item[100];
cin>>n;
if (n==0)
return NULL;
cin>>item;
TreeNode* root = new TreeNode(atoi(item));
count++;
queue<TreeNode*> nodeQueue;
nodeQueue.push(root);
while (count<n){
TreeNode* node = nodeQueue.front();
nodeQueue.pop();
cin>>item;
count++;
if (strcmp(item,"null")!=0){
int leftNumber = atoi(item);
node->left = new TreeNode(leftNumber);
nodeQueue.push(node->left);
}
if (count==n)
break;
cin>>item;
count++;
if (strcmp(item,"null")!=0){
int rightNumber = atoi(item);
node->right = new TreeNode(rightNumber);
nodeQueue.push(node->right);
}
}
return root;
}
int main(){
TreeNode* root;
root=inputTree();
int res=maxPathSum(root);
cout<<res<<endl;
}
输入说明 :
首先输入结点的数目n(注意,这里的结点包括题中的null空结点)
然后输入n个结点的数据,需要填充为空的结点,输入null。
输出说明 :
输出一个整数,表示结果。
输入范例 :
7
-10 9 20 null null 15 7
输出范例 :
42
int dfs(TreeNode* node,int &val){
if(node==NULL){
return 0;
}
int leftMax=max(0,dfs(node->left,val));//左子树中的最大值 如果为负数 直接舍弃
int rightMax=max(0,dfs(node->right,val));
int lmr=node->val+leftMax+rightMax;//left->root->right路径
int ret=node->val+max(leftMax,rightMax);//经过root的最大单边和
val=max(val,max(lmr,ret));//取上述二者最大值
return ret;//返回给上层路径的最大单边和
}
int maxPathSum(TreeNode* node){
if(node==NULL){
return 0;
}
int val=INT_MIN;
dfs(node,val);
return val;
}