第 21场双周赛:5339. 二叉搜索子树的最大键值和(DFS)

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

二叉搜索树的定义如下:

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

示例 1:

第 21场双周赛:5339. 二叉搜索子树的最大键值和(DFS)_第1张图片

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

第 21场双周赛:5339. 二叉搜索子树的最大键值和(DFS)_第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

 

你可能感兴趣的:(第 21场双周赛:5339. 二叉搜索子树的最大键值和(DFS))