AcWing 786. 第k个数

持续创作，加速成长！这是我参与「掘金日新计划 · 6 月更文挑战」的第5天，点击查看活动详情

AcWing 786. 第k个数

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

输入格式

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

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

输出格式

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

数据范围

1≤n≤100000

1≤k≤n

输入样例：

5 3
2 4 1 5 3

输出样例：

3

思路

将k值当做物理地址的值，比如第5个数其实就是数组4的位置，第2个数就是数组1的位置

每次只需要判断k在左区间还是右区间，一直递归查找k所在区间

最后只剩一个数时，只会有数组[k]一个数，返回数组[k]的值就是答案

ac代码

c++

#include<iostream>
using namespace std;
const int N =100001;
int arr[N];
int n,k;
int quick_sort(int l,int r,int k){
    if(l==r) return arr[l];
    int x = arr[(l+r)>>1],i = l-1,j = r+1;
    while(i<j){
        while(arr[++i]<x);
        while(arr[--j]>x);
        if(i<j)swap(arr[i],arr[j]);
    }
    int s = j-l+1;
    if(k<=s) return quick_sort(l,j,k);
    else quick_sort(j+1,r,k-s);
}
int main(){
    cin>>n>>k;
    for(int i=0;i<n;i++) cin>>arr[i];
    cout<<quick_sort(0,n-1,k)<<endl;
    return 0;
}

java

import java.io.*;
public class Main{
    public static void main(String [] args)throws IOException{
        BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
        String[] str = reader.readLine().split(" ");
        int n = Integer.parseInt(str[0]);
        int k = Integer.parseInt(str[1]);
        String[] str1 = reader.readLine().split(" ");
        int[] arr = new int [n];
        for(int i = 0;i < n;i++) arr[i] = Integer.parseInt(str1[i]);
        System.out.print(qucikSort(arr,0,n - 1,k));
        reader.close();
    }
    public static int qucikSort(int[] arr,int l,int r,int k){
        if(l >= r) return arr[l];
        int x = arr[l + r >> 1],i = l - 1,j = r + 1;
        while(i < j){
            while(arr[++i] < x);
            while(arr[--j] > x);
            if(i < j){
                int temp = arr[i];
                arr[i] = arr[j];
                arr[j] = temp;
            }
        }
        int s = j - l + 1;
        if(k <= s) return qucikSort(arr,l,j,k);
        else return qucikSort(arr,j + 1,r,k - s);
    }
}