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

给你一棵以 root 为根的 二叉树 ,请你返回 任意 二叉搜索子树的最大键值和。

二叉搜索树的定义如下:

任意节点的左子树中的键值都 小于 此节点的键值。
任意节点的右子树中的键值都 大于 此节点的键值。
任意节点的左子树和右子树都是二叉搜索树。
 

示例 1:

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

输入:root = [1,4,3,2,4,2,5,null,null,null,null,null,null,4,6]
输出:20
解释:键值为 3 的子树是和最大的二叉搜索树。
示例 2:

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

输入: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. 二叉搜索子树的最大键值和

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

上一篇:数据结构之栈


下一篇:Leetcode - 3. 无重复字符的最长子串