洛谷-求第k小的数题解

（1）使用c++sort排序+O2优化（不建议蒙混过关）

不开O2优化，后两个点会TLE
代码如下：

#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
int num[5000000];
int main()
{
    int n,k;
    cin>>n>>k;
    for(int i=0;i<n;i++)
        scanf("%d",&num[i]);
    sort(num,num+n);
    printf("%d",num[k]);
}

（2）c++强大的STL库

在强大的STL库中存在一个神奇的函数，那就是nth_element，这个函数主要用来将数组元素中第k小的整数排出来并在数组中就位，随时调用，可谓十分实用。

函数语句：nth_element(数组名,数组名+第k小元素,数组名+元素个数)
代码如下：

#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
int num[5000000];
int main()
{
    int n,k;
    cin>>n>>k;
    for(int i=0;i<n;i++)
        scanf("%d",&num[i]);
    nth_element(num,num+k,num+n);
    printf("%d",num[k]);
}

（3）快速排序 quick_sort 你值得拥有

快速排序点这里

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int num[5000005];
int n,k;
void quick_sort(int l,int r)
{
    if(l<r)
    {
        int v=num[(l+r)/2],i=l,j=r;//1 3 4 2 5
        while(1)
        {
            while(i<=j&&num[i]<v)i++;
            while(i<=j&&num[j]>v)j--;
            if(i<j)
            {
                swap(num[i++],num[j--]);
            }
            else
            break;
        }
        if(i==k)//i 就是第i小的数
        {
            cout<<num[k];
            exit(0);//结束程序
        }
        if(k<=j)//l-j  i i+1-r 分为三部分 中间的为第i小的数
        quick_sort(l,j);
        if(k>i)
        quick_sort(i,r);
    }
}
int main() {
	scanf("%d%d", &n,&k);
	k++;
	for(int i = 1; i <= n; i++)
    scanf("%d", &num[i]);
    quick_sort(1, n);
        
    return 0;
}