JAVA

import java.util.Scanner;

public class Main {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
int n = scanner.nextInt();
int k = scanner.nextInt();
int[] arr = new int[n];
for (int i = 0; i < n; i++) {
arr[i] = scanner.nextInt();
}
int kthSmallest = quickSelect(arr, 0, n - 1, k - 1); // 注意k-1，因为数组索引从0开始
System.out.println(kthSmallest);
}

private static int quickSelect(int[] arr, int left, int right, int k) {
    if (left == right) {
        return arr[left];
    }

    int pivotIndex = partition(arr, left, right);

    if (k == pivotIndex) {
        return arr[k];
    } else if (k < pivotIndex) {
        return quickSelect(arr, left, pivotIndex - 1, k);
    } else {
        return quickSelect(arr, pivotIndex + 1, right, k);
    }
}

// 这里使用了一个简单的划分策略，实际应用中可以根据需要优化
private static int partition(int[] arr, int left, int right) {
    int pivot = arr[right]; // 选择最右边的元素作为基准
    int i = left - 1;
    for (int j = left; j < right; j++) {
        if (arr[j] <= pivot) {
            i++;
            swap(arr, i, j);
        }
    }
    swap(arr, i + 1, right);
    return i + 1;
}

private static void swap(int[] arr, int i, int j) {
    int temp = arr[i];
    arr[i] = arr[j];
    arr[j] = temp;
}

}

在这个代码中，quickSelect函数是递归的核心，它接受一个数组arr，一个搜索范围的左边界left，一个右边界right，以及要查找的第k小元素的索引（注意这里索引是从0开始的，所以输入时的k需要减1）。partition函数负责将数组按照基准值进行划分，并返回基准值的新索引。最后，通过递归调用quickSelect，我们可以找到第k小的元素