刷题--二叉搜索树的第k个结点

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

基本思路：一个中序遍历的问题。

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回对应节点TreeNode
    def __init__(self):
        self.counter = 0
        self.p = None

    def KthNode(self, pRoot, k):
        # write code here
        if pRoot:
            self.KthNode(pRoot.left, k)
            self.counter += 1

            if self.counter == k:
                self.p = pRoot

            self.KthNode(pRoot.right, k)

            return self.p

也可以不设置p这样的变量，在两次递归调用的代码后面分别判断当前返回值是否为空，不为空就返回。