牛骨文教育服务平台(让学习变的简单)
博文笔记

第k小的数

创建时间:2018-03-07 投稿人: 浏览次数:182

总时间限制: 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);
}
声明:该文观点仅代表作者本人,牛骨文系教育信息发布平台,牛骨文仅提供信息存储空间服务。