LeetCode 230. 二叉搜索树中第K小的元素(递归)

题目描述

给定一个二叉搜索树，编写一个函数 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