题目描述
给定一棵二叉搜索树,请找出其中的第k小的结点。例如, (5,3,7,2,4,6,8) 中,按结点数值大小顺序第三小结点的值为4。
思路
中序遍历第k个数
class Solution:
# 返回对应节点TreeNode
def KthNode(self, pRoot, k):
# write code here
if not pRoot: return None
stack = []
while pRoot or stack:
while pRoot:
stack.append(pRoot)
pRoot = pRoot.left
pRoot = stack.pop()
k -= 1
if k == 0:
return pRoot
pRoot = pRoot.right