题目描述:
给你一棵以 root 为根的 二叉树 ,请你返回 任意 二叉搜索子树的最大键值和。
二叉搜索树的定义如下:
任意节点的左子树中的键值都 小于 此节点的键值。
任意节点的右子树中的键值都 大于 此节点的键值。
任意节点的左子树和右子树都是二叉搜索树。
示例 1:
输入:root = [1,4,3,2,4,2,5,null,null,null,null,null,null,4,6]
输出:20
解释:键值为 3 的子树是和最大的二叉搜索树。
示例 2:
输入:root = [4,3,null,1,2]
输出:2
解释:键值为 2 的单节点子树是和最大的二叉搜索树。
示例 3:
输入:root = [-4,-2,-5]
输出:0
解释:所有节点键值都为负数,和最大的二叉搜索树为空。
示例 4:
输入:root = [2,1,3]
输出:6
示例 5:
输入:root = [5,4,8,3,null,6,3]
输出:7
提示:
每棵树有 1 到 40000 个节点。
每个节点的键值在 [-4 * 10^4 , 4 * 10^4] 之间。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-sum-bst-in-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解析:
后序遍历:
1,子树不是二叉搜索树,则将根节点标记为9999(随意,大点就好)
2,判断父节点时,首先看子树是不是9999
若是99999,则子树为非二叉搜索树,自然当前根节点都不是二叉搜索树。
若不是99999,自然子树为二叉排序树,只需要找到左子树最大值和右子树最小值判断即可。
3,为二叉搜索树直接返回当前二叉搜索树的和即可。
代码 :
class Solution {
public:
//左侧为二叉搜索树的情况下,寻找左侧最大值
bool getLeft(TreeNode *root,int target){
while(root){
if(root->val>=target)
return true;
root=root->right;
}
return false;
}
//右侧为二叉搜索树的情况下,寻找右侧最大值
bool getRight(TreeNode *root,int target){
while(root){
if(root->val<=target)
return true;
root=root->left;
}
return false;
}
//传地址
int getMax(TreeNode* root,int &maxSum){
if(!root)
return 0;
//叶子结点,判断并返回
if(!root->left&&!root->right){
maxSum=max(maxSum,root->val);
return root->val;
}
//左右返回值(不为二叉搜索树返回0)
int left=getMax(root->left,maxSum);
int right=getMax(root->right,maxSum);
//左不为二叉搜索树返回0,并将根节点置为标记值(表明此节点为根下边不是二叉搜索树)
if(root->left&&(root->left->val==999999||getLeft(root->left,root->val)))
{
root->val=999999;
return 0;
}
//右不为二叉搜索树返回0,并将根节点置为标记值(表明此节点为根下边不是二叉搜索树)
if(root->right&&(root->right->val==999999||getRight(root->right,root->val)))
{
root->val=999999;
return 0;
}
//左右子树都是二叉搜索树
else {
maxSum=max(maxSum,root->val+left+right);
int res=root->val+left+right;
return res;
}
return 0;
}
int maxSumBST(TreeNode* root) {
int maxSum=0;
getMax(root,maxSum);
return maxSum;
}
};