给定一棵结点数为n 二叉搜索树,请找出其中的第 k 小的TreeNode结点值。 1.返回第k小的节点值即可 2.不能查找的情况,如二叉树为空,则返回-1,或者k大于n等等,也返回-1 3.保证n个节点的值不一样 数据范围: ,,树上每个结点的值满足 进阶:空间复杂度 ,时间复杂度 如输入{5,3,7,2,4,6,8},3时,二叉树{5,3,7,2,4,6,8}如下图所示: 该二叉树所有节点按结点值升序排列后可得[2,3,4,5,6,7,8],所以第3个结点的结点值为4,故返回对应结点值为4的结点即可。
class TreeNode:
def __init__(self, x):
self.val = x
self.left = None
self.right = None
def kthSmallest(root, k):
def inorder(node):
nonlocal k, result
if node:
inorder(node.left)
k -= 1
if k == 0:
result = node.val
return
inorder(node.right)
result = -1
inorder(root)
return result
# 构建示例树
root = TreeNode(5)
root.left = TreeNode(3)
root.right = TreeNode(7)
root.left.left = TreeNode(2)
root.left.right = TreeNode(4)
root.right.left = TreeNode(6)
root.right.right = TreeNode(8)
# 测试
print(kthSmallest(root, 3)) # 输出应为4