给你一棵以 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; } }