第k小元素

第k小元素

给定一个大小为n(1≤n≤1000000)且无序的整型数组，数组中可能存在相同元素，请找出该数组第k(1≤k≤n)小的元素，注意这里的第k小元素指的是按从小到大排序后的第k个位置上的元素。

输入格式:
每个输入文件为一个测试用例，每个文件的第一行给出两个正整数n和k，第二行给出n个整数，其间以空格分隔。

输出格式:
输出第k小元素的值。

输入样例:
10 4
2 3 5 12 4 9 3 8 2 9
输出样例:
3
代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB

思路：归并排序改写，找第k小，逐渐接近k的范围

代码：

#include <iostream>

using namespace std;
int a[1000100];
int n,k;
void findy(int l,int r)
{
    int temp=a[l];
    int i=l;
    int j=r;
    if(l>r)
        return;
    while(i<j)
    {
        while(j>i&&a[j]>temp)
            j--;
        a[i]=a[j];
        while(j>i&&a[i]<=temp)
            i++;
        a[j]=a[i];
    }
    a[i]=temp;
    if(i==k)
        cout<<a[k];
    else if(i>k)
        findy(l,i-1);
    else
        findy(i+1,r);
}
int main()
{
    cin>>n>>k;
    k=k-1;
    for(int i=0;i<n;i++)
        cin>>a[i];
    findy(0,n-1);
    return 0;
}

202205191605日