题目描述:
给定一个二叉搜索树,编写一个函数kthSmallest来查找其中第k个最小的元素。
说明:
你可以假设 k 总是有效的,1 ≤ k ≤ 二叉搜索树元素个数。
知识点:
二叉树中序遍历
思路和代码:
使用二叉树中序遍历的方法,找到第k小的数直接退出
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def kthSmallest(self, root: TreeNode, k: int) -> int:
temp_stack = [] # 存储栈
min_v = 0
while root or temp_stack:
while root:
temp_stack.append(root)
root = root.left
root = temp_stack.pop()
min_v = root.val
k-=1
if k==0:
break
root = root.right
return min_v