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

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

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

 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]
~~~~~
原文地址:https://www.cnblogs.com/missmzt/p/5552333.html