给定一个二叉搜索树,编写一个函数kthSmallest来查找其中第k个最小的元素。
对于二叉搜索树,前序 中序遍历就是一个有序的搜索,因此只需要在前序 中序遍历中记录一下搜索次数就行了,比较容易实现
public static int num = 0;
public static int remain = 0;
public int kthSmallest(TreeNode root, int k) {
remain = k;
visitK(root);
return num;
}
private void visitK(TreeNode root){
if(root == null){
return;
}
if(root.left != null){
visitK(root.left);
}
remain--;
if(remain == 0){
num = root.val;
return;
}
if(root.right != null){
visitK(root.right);
}
}