力扣练习——47 二叉树中的最大路径和
1.问题描述
给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。
示例 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>
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=Solution().maxPathSum(root);
cout<<res<<endl;
}
2.输入说明
首先输入结点的数目n(注意,这里的结点包括题中的null空结点)
然后输入n个结点的数据,需要填充为空的结点,输入null。
3.输出说明
输出一个整数,表示结果。
4.范例
输入
7
-10 9 20 null null 15 7
输出
42
5.代码
#include <iostream> #include <queue> #include <cstdlib> #include <cstring> 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 ans = INT_MIN;//初始值设置为最小 int dfs(TreeNode *root) { if (root == NULL) return 0; int left = dfs(root->left);//左子树提供的最大路径和 int right = dfs(root->right);//右子树提供的最大路径和 int interSum = root->val + left + right;//计算当前子树内部最大的路径和 ans = max(ans, interSum); int outerSum = root->val + max(0, max(left, right));//计算当前子树向外提供的最大值 return max(outerSum,0); } int maxPathSum(TreeNode *root) { //递归解决问题 dfs(root); return ans; } int main() { TreeNode* root; root = inputTree(); int res = maxPathSum(root); cout << res << endl; }

浙公网安备 33010602011771号