LeetCode1371

问题描述

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

二叉搜索树的定义如下：

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

思路

• DFS 递归
• 要判断该子树是否是二叉搜索树，需要四个条件：
  1.左子树是否是二叉搜索树
  2.右子树是否是二叉搜索树。
  3.根节点的值是否大于左子树最大值
  4.根节点的值是否小于右子树的最小值。

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def dfs(self,node):
    #返回 boolean(Node是否为BFS),val(子树和),max_left(BFS的最大值).min_right(BFS的最小值)
        if(node==None):
            return  True,0,float("-inf"),float("inf")
        l_bolean,l_val,l_max,l_min =  self.dfs(node.left)
        r_bolean,r_val,r_max,r_min =  self.dfs(node.right)
        
        if not l_bolean or not r_bolean:
            return False,-1,0,0
        # print(l_max,l_min,r_max,r_min)
        ## 判断当前节点是否属于二查子树
        # 当前节点必须大于左子树的最大值，小于右子树的最小值
        if node.val<=l_max or node.val>=r_min:
            return False,-1,0,0
        
        val = node.val+l_val+r_val
        if val>self.ans:
            self.ans = val
        ## 更新当前子树下的最大值和最小值
        if(node.left==None):
            node_min = node.val
        else:
            node_min = l_min
        if node.right==None:
            node_max = node.val
        else:
            node_max = r_max
        return True,val,node_max,node_min
    def maxSumBST(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        self.ans = 0
        self.dfs(root)
        return self.ans