JZ54 二叉搜索树的第k个节点

---

题意

JZ54 二叉搜索树的第k个节点

给定一棵结点数为n 二叉搜索树，请找出其中的第 k 小的TreeNode结点值。
1.返回第k小的节点值即可
2.不能查找的情况，如二叉树为空，则返回-1，或者k大于n等等，也返回-1
3.保证n个节点的值不一样

思路

中序排列：搜索二叉树的中序排列得到的序列是按节点值的升序排列； 二叉搜索树的第k个节点的值即为升序排列中下标为k-1对应的值。
注意：当k=0时，不存在第0个节点，返回-1。

代码

class Solution:
    def KthNode(self , proot: TreeNode, k: int) -> int:
        # write code here
        if not proot: return -1
        s = []
        def inorder(r):
            if not r: return
            inorder(r.left)
            s.append(r.val)
            inorder(r.right)
        inorder(proot)
        return s[k-1] if len(s)>=k else -1

---