二叉搜索子树的最大键值和

给你一棵以 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};
    }
};
posted @ 2023-05-21 23:00  失控D大白兔  阅读(11)  评论(0编辑  收藏  举报