题解 | #二叉搜索树的第k个节点#

#coding:utf-8
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param proot TreeNode类 
# @param k int整型 
# @return int整型
#
class Solution:
    def __init__(self):
        self.count = 0
    
    def KthNode(self , proot , k ):
        # write code here
        stack = []
        if k <=0:
            return -1
        cur = proot
        if not cur:
            return -1
        while cur or len(stack)>0:
            while cur:
                stack.append(cur)
                cur = cur.left
            while len(stack)>0:
                ret = stack.pop()
                self.count += 1
                print(ret.val)
                if self.count == k:
                    return ret.val
                
                if ret.right is not None:
                    cur = ret.right
#                     print(":",cur.val)
                    break
                else:
                    continue
        return -1