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

题目描述

给定一个二叉搜索树，编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。
说明：
你可以假设 k 总是有效的，1 ≤ k ≤ 二叉搜索树元素个数。

思路

详见链接

代码

class TreeNode:
	def __init__(self,x):
		self.val = x
		self.left = None
		self.right = None
	def kthSmallest(self,root:TreeNode, k:int) -> int:
		self.res, self.count = None, k
		def inorder(root):
			if not (root and self.count):
				return 
			inorder(root.left)
			self.count -= 1
			if not self.count:
				self.res = root.val
			inorder(root.right)
		inorder(root)
		return self.res