leetcode 之 Kth Smallest Element in a BST

题目描述：

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个小的元素。

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

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

class Solution(object):
def kthSmallest(self, root, k):
"""
:type root: TreeNode
:type k: int
:rtype: int
"""
if not root:
return -1

res = []
ls = []
res.append(root)
while len(ls) != k:
while root.left:
root = root.left
res.append(root)
root = res.pop()
ls.append(root.val)
if root.right:
root = root.right
res.append(root)
return ls[-1]