题目来源:98.验证二叉搜索树
题目:
给你一个二叉树的根节点 root ,判断其是否是一个有效的二叉搜索树。
有效二叉搜索树定义如下:
- 节点的左子树只包含 小于 当前节点的数。
- 节点的右子树只包含 大于 当前节点的数。
- 所有左子树和右子树自身必须也是二叉搜索树。
思路:中序遍历
看到左结点要小于根结点,右结点要大于根结点马上想到可能跟中序遍历有关,看了一下答案果真是,刷题又进步了一步,可以自己想出思路了,哦耶
题目所给的条件跟中序遍历的规则很相似,因此只要保证中序遍历后得到的数组是严格升序的即可,即五重复项,因此在最后的时候加了个set函数长度判断。具体代码如下:
# 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 isValidBST(self, root):
"""
:type root: TreeNode
:rtype: bool
"""
res = []
def InOrderTraverse(r):
if not r:
return
InOrderTraverse(r.left)
res.append(r.val)
InOrderTraverse(r.right)
InOrderTraverse(root)
if res == sorted(res) and len(res) == len(set(res)):
return True
else:
return False
不知道最后判断是不是还有更优的解法,感觉最后跑完时间和空间都不太优异,有好的想法的伙伴们可以一起讨论一下~
参考链接:Python验证二叉搜索树
LeetCode-Python-98. 验证二叉搜索树