98. 验证二叉搜索树

题目描述

给定一个二叉树,判断其是否是一个有效的二叉搜索树。

假设一个二叉搜索树具有如下特征:

  • 节点的左子树只包含小于当前节点的数。
  • 节点的右子树只包含大于当前节点的数。
  • 所有左子树和右子树自身必须也是二叉搜索树。

示例 1:

输入:
    2
   / \
  1   3
输出: true

示例 2:

输入:
    5
   / \
  1   4
     / \
    3   6
输出: false
解释: 输入为: [5,1,4,null,null,3,6]。
     根节点的值为 5 ,但是其右子节点值为 4 。

我的解法——递归求解(AC)

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def myisValidBST(self, root):
        if(not root):
            return True, None, None
            
        left_flag, left_min, left_max = self.myisValidBST(root.left)
        right_flag, right_min, right_max = self.myisValidBST(root.right)
        
        # 确定返回值
        res_min = root.val
        if(left_min and left_min < res_min):
            res_min = left_min
        if(right_min and right_min < res_min):
            res_min = right_min
        
        
        res_max = root.val
        if(left_max and left_max > res_max):
            res_max = left_max
        if(right_max and right_max > res_max):
            res_max = right_max
        
        # 确定flag
        if(left_max and left_max >= root.val):
            left_flag = False
        if(right_min and right_min <= root.val):
            right_flag = False
        
        return left_flag and right_flag, res_min, res_max
    
    
    def isValidBST(self, root):
        """
        :type root: TreeNode
        :rtype: bool
        """
        return self.myisValidBST(root)[0]