剑指offer——二叉搜索树的第K个节点（Python）

【题目】给定一棵二叉搜索树，请找出其中的第k小的结点。例如， （5，3，7，2，4，6，8） 中，按结点数值大小顺序第三小结点的值为4。
【思路】结合中序遍历和二叉搜索树的性质（二叉搜索树的左子树小于根节点，右子树大于根节点，所以中序遍历就是从小到大排好序的）得出中序序列后，输出第k个即可。
【代码】

class Solution:
    # 返回对应节点TreeNode
    def KthNode(self, pRoot, k):
        global res 
        res = []
        # write code here
        self.mid(pRoot)
        if k <= 0 or len(res) < k:
            return None
        else:
            return res[k-1]
    def mid(self,pRoot):
        if not pRoot:
            return None
        self.mid(pRoot.left)
        res.append(pRoot)
        self.mid(pRoot.right)