第k小的数

最新推荐文章于 2024-07-27 14:37:39 发布

Yuta233

最新推荐文章于 2024-07-27 14:37:39 发布

阅读量1k

点赞数 1

本文链接：https://blog.csdn.net/Yuta233/article/details/79473451

版权

总时间限制: 5000ms 单个测试点时间限制: 1000ms 内存限制: 65536kB 描述
给定包含N个（N <= 1000000）元素的数组a[1],a[2],a[3],…,a[N]，输出这些整数从小到大排序后的第k个（例如，k=1就是最小值）。

输入2行
第1行包含2个正整数N、k（1 <= k <= N <= 1000000），代表数组元素个数和要求第几小的数
第2行包含N个整数，空格隔开输出1个数，代表第k小数的大小样例输入8 3
5 2 3 7 1 3 2 6
样例输出2

#include <iostream>
using namespace std;
#define maxn 1000010
int a[maxn];
int n,k;

void quicksort(int leftEnd,int rightEnd,int k)
{
    if (leftEnd>rightEnd)
        return;
    int leftCursor=leftEnd;
    int rightCursor=rightEnd+1;
    int pivot=a[leftEnd];
    while(1)
    {
        do 
        {
            leftCursor++;
        } while (a[leftCursor]<=pivot&&leftCursor<rightEnd);
        do 
        {
            rightCursor--;
        } while (a[rightCursor]>pivot&&rightCursor>leftEnd); 

        if (leftCursor>=rightCursor)break;
        swap(a[rightCursor],a[leftCursor]);
    }   
    a[leftEnd]=a[rightCursor];
    a[rightCursor]=pivot;   

    int cnt=rightCursor-leftEnd;//the number of num in the left of rightcursor
    if (k==cnt+1)           
        cout<<a[rightCursor];
    else if(k<cnt+1)
        quicksort(leftEnd,rightCursor-1,k);//1 3 3
    else
        quicksort(rightCursor+1,rightEnd,k-cnt-1);
}

int main()
{
    //freopen("D:\\1.txt","r",stdin);
    cin>>n>>k;
    for (int i=1;i<=n;i++)  
        scanf("%d",a+i);
    quicksort(1,n,k);
}