码迷,mamicode.com
首页 > 其他好文 > 详细

leetcode 之 Kth Smallest Element in a BST

时间:2016-06-02 12:59:37      阅读:98      评论:0      收藏:0      [点我收藏+]

标签:

题目描述:

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]

 

leetcode 之 Kth Smallest Element in a BST

标签:

原文地址:http://www.cnblogs.com/missmzt/p/5552333.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!