题目描述
给定一个二叉搜索树,编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。 说明: 你可以假设 k 总是有效的,1 ≤ k ≤ 二叉搜索树元素个数。
思路
详见链接
代码
class TreeNode:def __init__(self,x):self.val = xself.left = Noneself.right = Nonedef kthSmallest(self,root:TreeNode, k:int) -> int:self.res, self.count = None, kdef inorder(root):if not (root and self.count):return inorder(root.left)self.count -= 1if not self.count:self.res = root.valinorder(root.right)inorder(root)return self.res