[剑指offer] 二叉搜索树的第k个结点

题目内容

给定一棵二叉搜索树，请找出其中的第k小的结点。例如， （5，3，7，2，4，6，8）    中，按结点数值大小顺序第三小结点的值为4。

https://www.nowcoder.com/practice/ef068f602dde4d28aab2b210e859150a?tpId=13&tqId=11215&tPage=4&rp=4&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

---

题目思路

对于一个二叉搜索树来说，我们使用中序遍历就可以得到一个从小到大的序列，从而求出第K个数字。

---

程序代码

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回对应节点TreeNode
    def KthNode(self, pRoot, k):
        # write code here
        if k<=0:
            return None
        mid=[]
        def midorder(pRoot):
            if not pRoot:
                return 
            midorder(pRoot.left)
            mid.append(pRoot)
            midorder(pRoot.right)
        midorder(pRoot)
        if k>len(mid):
            return None
        else:
            return mid[k-1]