快速排序 快速选择排序 Acwing 786第k个数

传送门

题目大意

基本上就是裸体
给定一个长度为n的整数数列，以及一个整数k，用快速选择算法求出数列的第k小的数是多少。

输入格式

第一行包含两个整数 n 和 k。
第二行包含 n 个整数（所有整数均在1~109109范围内），表示整数数列。

输出格式

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

数据范围

1≤n≤1000001≤n≤100000,

1≤k≤n

输入样例

5 3
2 4 1 5 3

输出样例

3

注意 在递归的时候一定要注意写上停止的条件 否则一直在循环不会停止 我经常忘记写出口

题解

方法一： 快速排序 o(nlogn)

#include<iostream>

using namespace std;

const int N = 1e5 + 10;

int n, a[N], k;

void quick_sort(int q[], int l, int r)
{
    if(l >= r) return ;
    int x = q[l + r >> 1], i = l - 1, j = r + 1;
    while(i < j)
    {
        do i ++ ; while(q[i] < x);
        do j -- ; while(q[j] > x);
        if(i < j) swap(q[i], q[j]);
    }
    quick_sort(q, l, j);
    quick_sort(q, j + 1, r);
}

int main ()
{
    scanf("%d %d",&n, &k);
    for(int i = 1; i <= n; i ++ )
    {
        scanf("%d", &a[i]);
    }
    quick_sort(a, 1, n);
    cout << a[k] << endl;
    return 0;
}

方法二： 快速选择排序 o(n)

#include<iostream>

using namespace std;

const int N = 1e5 + 10;

int n, k;
int a[N];

int quick_sort(int q[], int l, int r, int k)
{
    if(l == r) return q[l];
    int x = q[l + r >> 1], i = l - 1, j = r + 1;
    while(i < j)
    {
        do i ++ ; while(q[i] < x);
        do j -- ; while(q[j] > x);
        if(i < j) swap(q[i], q[j]);
    }
    int sl = j - l + 1;
    if(k <= sl) return quick_sort(q, l, j, k);
    return quick_sort(q, j + 1, r, k - sl);
}

int main ()
{
    scanf("%d%d",&n, &k);
    for(int i = 1; i <= n; i ++ )
    {
        scanf("%d", &a[i]);
    }
    cout << quick_sort(a, 1, n, k) << endl;
    return 0;
}