leetcode 之 Kth Smallest Element in a BST

Posted 2020-07-19 鱼儿慢慢游～～

题目描述：

Given a binary search tree, write a function kthSmallest to find the kth smallest element in it.

Note:
You may assume k is always valid, 1 ≤ k ≤ BST‘s total elements.

给定一棵二叉查找树， 找到第k个小的元素。

思路，利用二叉查找树的特性，其左子树的元素小于根节点，右子树的元素大于根节点。当采用中序遍历时，可以得到从小到大的排列。

因此采用中序遍历代码如下：

 1 class Solution(object):
 2     def kthSmallest(self, root, k):
 3         """
 4         :type root: TreeNode
 5         :type k: int
 6         :rtype: int
 7         """
 8         if not root:
 9             return -1
10         
11         res = []
12         ls = []
13         res.append(root)
14         while len(ls) != k:
15             while root.left:
16                 root = root.left
17                 res.append(root)
18             root = res.pop()
19             ls.append(root.val)
20             if root.right:
21                 root = root.right
22                 res.append(root)
23         return ls[-1]