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

算法题（程序员面试宝典）

解题思路主要来源于leetcode官方与《程序员面试宝典》&labuladong

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

给你一棵以 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] 之间。

解题方法

解题思路1

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    int maxSum = Integer.MIN_VALUE;
    public int maxSumBST(TreeNode root) {
        //合法性判断
        boolean flag = isNotValid(root);
        if(flag==true){
            //全部为负数
            return 0;
        }
        traverse(root);
        return maxSum;
    }


    //int[]{isBST,maxValue,minValue,sum}
    //isBST: 1 -> 是BST树　　　0 -> 不是BST树
    //maxValue： 右子树的最大值
    //minValue:  左子树的最小值
    //sum:   当前BST树的键值和
    public int[] traverse(TreeNode root){
        //要想得到最大的键值和，1，判断当前根的左右子树是否为BST  2,当左右子树为BST时，要知道左子树的最大值，右子树的最小值，来判断以当前节点
        //续：root（根据BST树的定义）为根的树是否为BST  3，若当前节点root的树为BST 则计算其最大键值和(左子树值+右子树的值+root值)
        //由上述可知，要想得到最大键值和，首先要知道左右子树的值，所以对二叉树采用后序遍历，
        //base case
        if(root==null){
            return new int[]{1,Integer.MIN_VALUE,Integer.MAX_VALUE,0};
        }

        int[] left = traverse(root.left);
        int[] right = traverse(root.right);
    
        //首先判断以当前节点为根的树是否是BST树
        int[] res = new int[4];
        if(left[0]==1&&right[0]==1&&root.val<right[2]&&root.val>left[1]){
            //以当前节点为根的树是BST树
            res[0]=1;
            //当前树的最大值
            res[1]=Math.max(right[1],root.val);
            //当前树的最小值
            res[2]=Math.min(left[2],root.val);
            //计算当前树键值和
            res[3]=left[3]+right[3]+root.val;
            //保存当前的最大键值和
            maxSum = Math.max(maxSum,res[3]);
        }else{
            res[0] = 0;
        }
        return res;    
    }

    public boolean isNotValid(TreeNode root){
        //true:全为负数  false:有正数
        //遍历二叉树
        if(root==null){
            return true;
        }
        boolean left = isNotValid(root.left);
        boolean right = isNotValid(root.right);
        if(left&&right&&root.val<0){
            return true;
        }
        return false;
    }
}