3、Acwing:第k个数

3、2021/1/18 第k个数

786. 第k个数

给定一长度为n的整数数列，以及一个整数k，请用快速选择算法求出数列从小到大排序后的第k个数。

输入格式

第一行包含两个整数 n 和 k。

第二行包含 n 个整数（所有整数均在1~109109范围内），表示整数数列。

输出格式

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

数据范围

1≤n≤1000001≤n≤100000,
1≤k≤n1≤k≤n

输入样例：

5 3
2 4 1 5 3

输出样例：

3

难度：简单
时/空限制：1s / 64MB
总通过数：17757
总尝试数：29314
来源：模板题,AcWing
算法标签

import java.util.*;
public class Main{
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int k = sc.nextInt(); //要找的第 k 个数
        int[] q = new int[100010];
        for(int i = 0 ; i < n ; i++){
            q[i] = sc.nextInt();
        }
        
        quick_sort(q , 0 , n  - 1);
        
        System.out.println(q[k-1]); //因为下标是从0开始，所以要减一
    }
    
    public static void quick_sort(int[] q , int l , int r){
        if(l >= r){
            return;
        }
        
        int i = l - 1;
        int j = r + 1;
        int x = q[l + r >> 1];
        
        while(i < j){
            do i++; while(q[i] < x);
            do j--; while(q[j] > x);
            if(i < j){
                int t = q[i];
                q[i] = q[j];
                q[j] = t;
            }
        }
        quick_sort(q , l , j);
        quick_sort(q , j + 1 , r);
    }
}