第k个数—java代码

给定一个长度为 n 的整数数列，以及一个整数 k，请用快速选择算法求出数列从小到大排序后的第 k 个数。
输入格式
第一行包含两个整数 n 和 k。
第二行包含 n 个整数（所有整数均在 1∼10⁹ 范围内），表示整数数列。

输出格式
输出一个整数，表示数列的第 k 小数。

数据范围
1≤n≤100000,
1≤k≤n

输入样例 ：

5 3
3 1 2 4 5

输出样例 ：

3

import java.util.*;
public class Main {
    private static int N = 100010;
    private static int[] q = new int[N];
    
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int num = sc.nextInt();
        int k = sc.nextInt();
        for (int i = 0; i < num; i ++) {
            q[i] = sc.nextInt();
        }
        
            System.out.print(quick_sort(0, num - 1, k));
        
    }
    
    public static int quick_sort(int l, int r, int k) {
        if (l == r) return q[l]; 
        int i = l - 1;
        int j = r + 1;
        int x = q[l + r >> 1];
        
        while (i < j) {
            while(q[++ i] < x);
            while(q[-- j] > x);
            if (i < j) {
            int temp = q[i];
            q[i] = q[j];
            q[j] = temp;
            }
        }
        int sl = j - l + 1;
        if (k <= sl) return quick_sort(l, j, k);
        
        return quick_sort(j + 1, r , k - sl);
	}
}