P2-2017级算法第二次上机 C 第k顺序统计量

题目描述

给定3个整数A, B, C, 和数组第一个数a[1]。数组a由如下方式得到。询问a中第k小的数是多少。

for(int i = 2; i <= 3000000; ++i)

a[i] = ((1LL * a[i - 1] * A ^ B) + C) % 1000000007;

输入

多组组数据，每组数据一行，数据组数不超过7

每行5个整数A, B, C, a[1], k (保证五个数为int范围内正整数且k在3000000范围内)

输出

对于每组数据，输出一行，第k小的数。

输入样例

4 5 6 7 8

输出样例

972

思路

第k顺序统计量查找，推荐使用STL的nth_element通过，选择分治的思想也可以，但是直接进行排序是一定无法通过的。时间复杂度O(n)。

关于nth_element：

头文件：<algorithm>

作用：nth_element作用为求第n大的元素，并把它放在第n位置上，下标是从0开始计数的，也就是说求第0小的元素就是最小的数。

还不是特别清楚，亟待解决。

参考代码

#include<cstdio>  
#include<algorithm>  
const int MAXN = 3000002;  
int a[MAXN],A,B,C,k;  
using namespace std;  
int main()  
{  
    while(scanf("%d%d%d%d%d",&A,&B,&C,&a[1],&k) == 5){  
        for(int i = 2;i <= 3000000;i++)  
            a[i] = ((1LL * a[i-1] * A ^ B)+C)%1000000007;  
        nth_element(a+1,a+k,a+3000001);  
        printf("%d\n",a[k]);  
    }  
}