二叉搜索树的第k个节点

题目描述
给定一棵二叉搜索树,请找出其中的第k小的结点。例如, (5,3,7,2,4,6,8) 中,按结点数值大小顺序第三小结点的值为4。

class Solution:
    def __init__(self):
        self.array = []
    # 返回对应节点TreeNode
    def KthNode(self, pRoot, k):
        if not pRoot or k <= 0:
            return None
        self.midTraversal(pRoot, k)
        if len(self.array) >= k:
            return self.array[k - 1]
        return None

    def midTraversal(self, pRoot, k):
        if len(self.array) >= k or not pRoot:
            return
        self.midTraversal(pRoot.left, k)
        self.array.append(pRoot)
        self.midTraversal(pRoot.right, k)

你可能感兴趣的:(剑指offer,树)