剑指Offer 54 二叉搜索树的第k大节点

二叉搜索树的第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:
    def __init__(self):
        self.l = []
        self.count = 0
        
    def inOrder(self,root,k):
        if root != None:
            self.inOrder(root.left,k)
            self.count += 1
            if self.count == k:
                self.l.append(root)
            self.inOrder(root.right,k)
        
    # 返回对应节点TreeNode
    def KthNode(self, pRoot, k):
        if k == 0:
            return None
        self.inOrder(pRoot,k)
        if k > self.count:
            return None
        return self.l[0]
        # write code here