算法日记本 | LeetCode 98. 验证二叉搜索树

103 阅读 0 评论 68 点赞

我是靠谱客的博主清秀裙子，这篇文章主要介绍算法日记本 | LeetCode 98. 验证二叉搜索树，现在分享给大家，希望可以做个参考。

题目描述

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

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

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

示例 1:

复制代码

1
2
3
4
5
6
7
输入:
2
/ 
1
3
输出: true

示例 2:

复制代码

1
2
3
4
5
6
7
8
9
10
11
12
输入:
5
/ 
1
4
/ 
3
6
输出: false
解释: 输入为: [5,1,4,null,null,3,6]。
根节点的值为 5 ，但是其右子节点值为 4 。

分析解答

乍一看是道挺简单的题目，二叉搜索树的一个特点是：中序遍历得到的结果是一个递增的数组，直接利用这个结论，通过递归实现一个中序遍历，再遍历一下得到的数组即可：

复制代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
class Solution(object):
def __init__(self):
self.arr = list()
def isValidBST(self, root):
"""
:type root: TreeNode
:rtype: bool
"""
self.middleOrder(root)
for i in range(len(self.arr) - 1):
if self.arr[i] >= self.arr[i + 1]:
return False
return True
def middleOrder(self, root):
if root is None:
return
if root.left is None:
self.arr.append(root.val)
self.middleOrder(root.right)
elif root.right is None:
self.middleOrder(root.left)
self.arr.append(root.val)
else:
self.middleOrder(root.left)
self.arr.append(root.val)
self.middleOrder(root.right)

提交结果，验证通过！不过看了眼结果，只超过了7%的提交，有点小忧伤，看来这种方法性能还是比较差的，主要的原因应该有两个：

有些时候其实很早就可以得出否定的结论，比如根节点的值就小于它左边子节点的值，但这种方法一定要将全部节点值遍历之后才能得出结果，增加了平均的时间复杂度。
对数组进行遍历增加了额外的时间开销，实际上可以在递归的过程中一起进行判断，只要我们维护一个当前最后访问到的节点的值即可，这样在遍历时就可以保证有序性，或者在中途不满足条件时直接得到否定的结论。

按照这种思路，可以得到一个更优的中序遍历的实现：

复制代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
import sys
class Solution(object):
def __init__(self):
self.last = sys.float_info.max * -1
def isValidBST(self, root):
"""
:type root: TreeNode
:rtype: bool
"""
if root is None:
return True
if self.isValidBST(root.left):
if self.last < root.val:
self.last = root.val
return self.isValidBST(root.right)
return False