leetcode 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] 之间。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/maximum-sum-bst-in-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

前序遍历 + 用数组记录每个节点的状态

1：首先判断是不是搜索树，再计算和。

2：递归前序遍历二叉树，返回值为一个长度为5数组arr

arr[0] 当前树的最小值

arr[1] 当前树的最大值

arr[2] 当前树是否为搜索树，0：不是，1：是

arr[3] 当前树的若是搜索树，则其和

arr[4] 当前树的子搜索树的最大和，若为负数，则赋值为0

3：根据左右节点的返回值，组装当前节点的返回值。

4：最后返回arr[4]即为答案。

/**
 * 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 {
   public int maxSumBST(TreeNode root) {
        int[] ints = find(root);
        return ints[4];
    }

    public int[] find(TreeNode node) {
        int val = node.val;
        if (node.left == null && node.right == null) {
            return new int[]{val, val, 1, val, val > 0 ? val : 0};
        }
        int[] al = null;
        int[] ar = null;
        if (node.left != null) {
            al = find(node.left);
        }
        if (node.right != null) {
            ar = find(node.right);
        }
        if (ar == null) {
            if (al[2] == 0) {
                return al;
            }
            if (al[1] >= val) {
                al[2] = 0;
                return al;
            }
            al[1] = val;
            al[3] += val;
            al[4] = Math.max(al[3], al[4]);
            return al;
        }
        if (al == null) {
            if (ar[2] == 0) {
                return ar;
            }
            if (ar[0] <= val) {
                ar[2] = 0;
                return ar;
            }
            ar[0] = val;
            ar[3] += val;
            ar[4] = Math.max(ar[3], ar[4]);
            return ar;
        }
        if (al[2] == 0 || ar[2] == 0 || al[1] >= val || ar[0] <= val) {
            al[2] = 0;
            al[4] = Math.max(al[4], ar[4]);
            return al;
        }
        al[1] = ar[1];
        al[3] = al[3] + ar[3] + val;
        al[4] = Math.max(al[3], Math.max(al[4], ar[4]));
        return al;
    }
}

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

推荐阅读