一、题目描述
给定一棵二叉搜索树,请找出其中的第k小的结点。例如, (5,3,7,2,4,6,8) 中,按结点数值大小顺序第三小结点的值为4。
二、解题思路
二叉搜索树按照中序遍历的顺序打印出来正好就是排序好的顺序。
所以,按照中序遍历顺序找到第k个结点就是结果。
⚠️:要注意,返回的是节点,而不是节点的值!!!
三、Code(python)
中序遍历(递归):
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
# 返回对应节点TreeNode
def KthNode(self, pRoot, k):
# write code here
global res
res = []
self.KthNode2(pRoot)
if k<=0 or k > len(res):
return None
else:
return res[k-1]
def KthNode2(self, pRoot):
if not pRoot:
return None
self.KthNode2(pRoot.left)
res.append(pRoot)
self.KthNode2(pRoot.right)