二叉树——13.验证二叉搜索树

力扣题目链接

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

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

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

二叉搜索树性质：

首先，需要理解二叉搜索树的基本性质：

1. 对于每一个节点，左子树中的所有节点的值都小于该节点的值。
2. 对于每一个节点，右子树中的所有节点的值都大于该节点的值。
3. 二叉搜索树中没有重复的元素。

基于这个性质，如果我们对一棵二叉搜索树进行中序遍历，我们会得到一个严格递增的有序数组。这一特性是解决问题的关键。

解题的基本思路

1. 中序遍历：对给定的二叉树进行中序遍历，将遍历的结果按顺序存储在一个列表中。

   • 中序遍历的顺序是：先遍历左子树，再访问根节点，最后遍历右子树。
   • 对于二叉搜索树，中序遍历的结果应该是一个从小到大排列的有序数组。

2. 检查有序性：通过遍历这个结果列表，检查它是否是严格递增的。

   • 如果发现任意两个相邻的元素不满足严格递增（即前一个元素大于或等于后一个元素），那么这棵树就不是一个有效的二叉搜索树，直接返回 False。
   • 如果遍历完整个列表，没有发现任何违反严格递增的情况，那么可以确定这是一棵有效的二叉搜索树，返回 True。

完整代码如下：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def __init__(self):
        self.vec = []

    def traversal(self, root):
        if root is None:
            return
        self.traversal(root.left)
        self.vec.append(root.val)  # 将二叉搜索树转换为有序数组
        self.traversal(root.right)

    def isValidBST(self, root):
        self.vec = []  # 清空数组
        self.traversal(root)
        for i in range(1, len(self.vec)):
            # 注意要小于等于，搜索树里不能有相同元素
            if self.vec[i] <= self.vec[i - 1]:
                return False
        return True

• Solution 是一个包含验证二叉搜索树功能的类。
• __init__ 方法是这个类的构造函数。在初始化时，创建了一个空列表 self.vec，这个列表将用来存储二叉搜索树中序遍历的结果。

def traversal(self, root):
    if root is None:
        return
    self.traversal(root.left)
    self.vec.append(root.val)  # 将二叉搜索树转换为有序数组
    self.traversal(root.right)

• traversal 方法是一个递归函数，用来对二叉树进行中序遍历。
• 中序遍历的顺序是：先遍历左子树，然后访问根节点，最后遍历右子树。对于二叉搜索树来说，中序遍历的结果应该是一个从小到大排序的数组。
• 具体步骤如下：
  1. 如果当前节点 root 是空的（None），说明递归到达了叶子节点的子节点，直接返回。
  2. 递归调用 traversal(root.left)，遍历左子树。
  3. 将当前节点的值 root.val 添加到 self.vec 列表中。
  4. 递归调用 traversal(root.right)，遍历右子树。

通过这个方法，self.vec 将存储二叉树所有节点的值，并且按照中序遍历的顺序排列，这在二叉搜索树中应该是一个升序的数组。

def isValidBST(self, root):
    self.vec = []  # 清空数组
    self.traversal(root)
    for i in range(1, len(self.vec)):
        # 注意要小于等于，搜索树里不能有相同元素
        if self.vec[i] <= self.vec[i - 1]:
            return False
    return True

• isValidBST 方法用于判断给定的二叉树 root 是否是一棵有效的二叉搜索树。
• 首先，它会清空 self.vec 列表，以确保不受上一次遍历结果的影响。
• 然后调用 traversal 方法，对传入的二叉树进行中序遍历，并将结果存储在 self.vec 列表中。
• 接下来，使用一个 for 循环，从列表的第二个元素（索引为1）开始，逐个与前一个元素进行比较：
  • 如果发现当前元素 self.vec[i] 小于或等于前一个元素 self.vec[i - 1]，那么就说明这个二叉树不是一个有效的二叉搜索树，因为在二叉搜索树中，中序遍历结果必须是严格递增的（即没有相等的元素）。
  • 如果发现了不满足条件的情况，函数立即返回 False，表示这不是一棵有效的二叉搜索树。
• 如果整个循环结束后，没有发现任何不满足条件的情况，则返回 True，表示这棵二叉树是一棵有效的二叉搜索树。