hdu6040
题意
将一个函数运行 n 次,一共得到 n 个值,有 m 次询问,每次询问第 k 小的值。
分析
考察了 (nth\_element) 函数的运用。(nth\_element(a, a + x, a + n)) 使得 ( (a) 数组下标范围 ([0, n)) ) (a[x]) 存的是第 (x) 小的数,且保证 (x) 左边的数小于等于它,右边的数大于等于它( 当 (x = 0) 时, (a[0]) 是最小的值 )。
将 (k) 排序后从大到小,求值,并更新 (n) 的值,因为右边的数一定大于等于左边的数,剩下第 k 小的取值一定能在左边取到。
code
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<iostream>
using namespace std;
typedef long long ll;
const int MAXN = 1e7 + 10;
int kase = 1;
int n, m;
unsigned A, B, C;
unsigned x, y, z;
unsigned rng61() {
unsigned t;
x ^= x << 16;
x ^= x >> 5;
x ^= x << 1;
t = x;
x = y;
y = z;
z = t ^ x ^ y;
return z;
}
unsigned a[MAXN];
struct B {
int x, i;
bool operator<(const B&other) const {
return x > other.x;
}
}b[105];
unsigned ans[105];
int main() {
while(~scanf("%d%d%u%u%u", &n, &m, &A, &B, &C)) {
x = A; y = B; z = C;
for(int i = 0; i < n; i++) {
a[i] = rng61();
}
for(int i = 0; i < m; i++) {
scanf("%d", &b[i].x);
b[i].i = i;
}
sort(b, b + m);
int len = n;
for(int j = 0; j < m; j++) {
if(j && b[j].x == b[j - 1].x) {
ans[b[j].i] = ans[b[j - 1].i];
continue;
}
nth_element(a, a + b[j].x, a + len);
ans[b[j].i] = a[b[j].x];
len = b[j].x;
}
printf("Case #%d:", kase++);
for(int i = 0; i < m; i++) {
printf(" %u", ans[i]);
}
printf("
");
}
return 0;
}