题解 | #二叉搜索树的第k个节点#
二叉搜索树的第k个节点
http://www.nowcoder.com/practice/57aa0bab91884a10b5136ca2c087f8ff
递归方法找中序遍历的第k位元素。
# 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.res=-1
self.count=0
def inorder(self, root:TreeNode,k):
if not root or self.count>k:#递归终止条件:已经遍历完或者遍历数超过k时
return self.res
else:
self.inorder(root.left,k)
self.count=self.count+1
if self.count==k:
self.res=root.val#只有到这一步才是真正在遍历寻找
self.inorder(root.right,k)
def KthNode(self , proot: TreeNode, k: int) -> int:
self.inorder(proot,k)
return self.res