给你一棵以 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
提示:
每棵树最多有 40000 个节点。
每个节点的键值在 [-4 * 10^4 , 4 * 10^4] 之间。
思路:深搜遍历整棵树,开两个map标记当前节点为根的子树是否是BST,另一个用来保存该子树的和。
class Solution {
private int ans;
private Map map1;
private Map map2;
public int maxSumBST(TreeNode root) {
ans=0;
map1=new HashMap<>();
map2=new HashMap<>();
dfs(root);
return ans;
}
private void dfs(TreeNode root) {
if(root==null) return;
dfs(root.left);
dfs(root.right);
if(map2.containsKey(root)) return;
if(jud(root))
map1.put(root, findSum(root));
}
private boolean jud(TreeNode root) {
if(root==null) return true;
if(map2.containsKey(root))
return map2.get(root);
boolean flag1=jud(root.left);
boolean flag2=jud(root.right);
if(flag1 && flag2) {
int x=findMax(root.left);
int y=findMin(root.right);
if(x