二叉搜索子树的最大键值和
给你一棵以 root 为根的 二叉树 ,请你返回任意二叉搜索子树的最大键值和
1. 中序遍历+记录多个返回值
class Solution {
public:
//求子树键值需要后序遍历,判断二叉搜索树需要中序遍历
//判断二叉树也可以用判断左右值加中序遍历的方式,所以这里采用中序遍历
int res = 0;
int maxSumBST(TreeNode* root) {
vector<int> ans = dfs(root);
return res;
}
vector<int> dfs(TreeNode* root){//返回一个四元组,分别为后面所需要的子树最小值、最大值、键值和、是否是二叉搜索树
if(!root) return {1,INT_MAX,INT_MIN,0};
vector<int> left = dfs(root->left);
vector<int> right = dfs(root->right);
bool flag = false;
if(left[0]&&right[0]&&root->val>left[2]&&root->val<right[1]) flag = true;
if(!flag) return {0,INT_MAX,INT_MIN,0};//不是二叉搜索树就无需再计算了
int cur = root->val+left[3]+right[3];//键值和
res = max(res,cur);
int mi = min({root->val,left[1],right[1]});
int mx = max({root->val,left[2],right[2]});
return {1,mi,mx,cur};
}
};