题目描述:
给定一个二叉搜索树的根节点 root ,和一个整数 k ,请你设计一个算法查找其中第 k 个最小元素(从 1 开始计数)。
示例:
算法思想:
借助二叉搜索树得天独厚的优势–典型的中序遍历思想
我们知道中序遍历可以从小到大的打印二叉搜索树
traverse(root.left);
print(root.val);
traverse(root.right);
借助这样的思想,我们可以用一个rank变量做排名,和打印一样,从小到大,当排到k时,返回输出即可。
代码实现:
class Solution {
public int kthSmallest(TreeNode root, int k) {
traverse(root, k);
return res;
}
//记录结果值
int res = 0;
//记录当前元素排名
int rank = 0;
public void traverse(TreeNode root, int k) {
if(root == null) return;
//访问左节点
traverse(root.left, k);
//遍历根
rank++;
if(rank == k) {
res = root.val;
return;
}
//访问右节点
traverse(root.right, k);
}
}