题目描述:给定一个二叉搜索树的根节点 ,和一个整数 ,请你设计一个算法查找其中第 个最小元素(从 1 开始计数)。
获得更多?算法思路:代码文档,算法解析的私得。
运行效果
完整代码
/**
* 2 * @Author: LJJ
* 3 * @Date: 2023/8/21 13:31
* 4
*/
public class KthSmallestElement {
static class TreeNode{
int val;
TreeNode left;
TreeNode right;
TreeNode(int val){
this.val = val;
}
}
private int count = 0; // 计数器,用于记录已经访问的节点个数
private int result = 0; // 存储第 k 个最小元素的值
public int kthSmallest(TreeNode root, int k){
inorderTraversal(root,k); //开始中序遍历查找第k个最小元素
return result;
}
private void inorderTraversal(TreeNode node, int k){
if (node == null || count >= k){
return; //诋毁终止条件,节点为空或计数器达到k
}
inorderTraversal(node.left, k); // 递归遍历左子树
count++; //计数器加一
if (count == k){
result = node.val; // 如果计数器等于 k,说明找到第 k 个最小元素
}
inorderTraversal(node.right, k); // 递归遍历右子树
}
public static void main(String[] args) {
// 构建二叉搜索树
TreeNode root = new TreeNode(5);
root.left = new TreeNode(2);
root.left.right = new TreeNode(4);
root.right = new TreeNode(9);
root.right.left = new TreeNode(6);
root.right.right= new TreeNode(10);
KthSmallestElement solution = new KthSmallestElement();
int k = 3; // 要找第 k 个最小元素
int result = solution.kthSmallest(root, k);
System.out.println("第 " + k + " 个最小元素是: " + result); // 输出结果
}
}