给定一个二叉搜索树的根节点 root
,和一个整数 k
,请你设计一个算法查找其中第 k
个最小元素(从 1 开始计数)。
示例 1:
输入:root = [3,1,4,null,2], k = 1
输出:1
示例 2:
输入:root = [5,3,6,2,4,null,null,1], k = 3
输出:3
提示:
- 树中的节点数为
n
。 1 <= k <= n <= 104
0 <= Node.val <= 104
public class BinarySearchTreeKthSmallestNodeSolution {
private int result;
private int count = 0;
public int kthSmallest(TreeNode root, int k) {
inorderTraversal(root, k);
return result;
}
// 二叉搜索树中序遍历的序列是一个递增序列
private void inorderTraversal(TreeNode root, int k) {
if (root == null) {
return;
}
// 如果已经找到了,后续递归直接返回
// 但还是会遍历完整棵树
if (count == k) {
return;
}
inorderTraversal(root.left, k);
if (++count == k) {
result = root.val;
return;
}
inorderTraversal(root.right, k);
}
}