分治法——求解第k小问题(线性时间选择)

第k个数

给定一个长度为 n 的整数数列，以及一个整数 k，请用快速选择算法求出数列从小到大排序后的第 k个数。

输入格式

第一行包含两个整数 n 和 k。

第二行包含 n 个整数（所有整数均在 1∼10^9 范围内），表示整数数列。

输出格式

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

数据范围

1≤n≤100000,
1≤k≤n

输入样例：

5 3
2 4 1 5 3

输出样例：

3

老师提了一个问题：对于选择第k小问题，线性时间选择算法选择五个元素分组来查找中位数的中位数，请问为什么要选择五个元素一组呢？

思路：利用快排划分的思想，取五个元素为一组，然后取中位数的中位数，减少极端数据的影响
(快排最坏时间复杂度O(n^2)）

#include<iostream>
#include<algorithm>

using namespace std;
const int inf = 0x3f3f3f3f;
const int N = 1e7 + 10;
int n, a[N], pos, k;
int paration(int a[], int l, int r, int k)//快排
{
    for(int i = l;i <= r;i++)
        if(a[i] == k) {swap(a[i], a[l]);break;}
    int i = l, j = r + 1;
    while(i < j) 
    {
        do i++; while(a[i] < k);
        do j--; while(a[j] > k);
        if(i < j) swap(a[i], a[j]);
    }
    swap(a[l], a[j]);
    return j;
}
int select(int a[], int l, int r, int k) {
    //在数组a[l...r]中选择第k小的元素返回
    if(r - l < 5) {
        sort(a + l, a + r + 1);
        return a[l + k - 1];
    }
    int pos = (r - l + 1) / 5;
    //每组排序后，将中位数序列置于整个数组的前部分
    for(int i = 0;i < pos;i++)
    {
        int s = l + 5 * i;
        int t = s + 4;
        sort(a + s, a + t + 1);//将一组三个元素进行排序
        swap(a[l + i], a[s + 2]);//将第i组的中位数放置在原数组的第(l + i)位置
    }
    // for(int i = l;i <= pos;i++) printf("%d ", a[i]);
    //选取中位数序列的中位数
    int x = select(a, l, l + pos - 1, (pos + 1) / 2);
    int i = paration(a, l, r, x);
    int j = i - l + 1;
    if(k == j) return a[i];
    if(k < j) return select(a, l, i, k);
    else return select(a, i + 1, r, k - j);
}

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