hdu 6040 Hints of sd0061[2017 Multi-University Training Contest - Team 1] [STL]

题目链接

题意: 求a数组的第b[i]大数

分析: (1)首先要了解c++中求第k大数的STL函数nth_element(start,start + n, end); 这个函数可以使数组第n大元素处于第n位置（从0开始,其位置是下标为 n的元素），并且比这个元素小的元素都排在这个元素之前，比这个元素大的元素都排在这个元素之后，但不能保证他们是有序的. 复杂度为O(n).

(2) 将b数组从大到小排序, 由于由于排在前面的都是比第b[i]小元素, 所以下次调用函数中end参数, 就是上次得到的下标. 

#include<cstdio>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn = 1e7 + 10;
const ll mod = 1e9 + 7;

struct G {
    int x, i;
    bool operator < (const G &t) const {
        return x > t.x;
    }
};

unsigned gg[maxn], ans[120];
G r[120];
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;
}

int main() {
    int v = 1, n, m;
    unsigned a, b, c;
    while(~scanf("%d %d %u %u %u", &n, &m, &a, &b, &c)) {
        for(int i = 0; i < m; i++) {
            scanf("%d", &r[i].x);
            r[i].i = i;
        }
        x = a; y = b; z = c;
        for(int i = 0; i < n; i++) {
            gg[i] = rng61();
        }
        sort(r, r + m);
        int nn = n;
        for(int i = 0; i < m; i++) {
            if(i && r[i].x == r[i - 1].x) {
                ans[r[i].i] = ans[r[i - 1].i];
                continue;
            }
            nth_element(gg, gg + r[i].x, gg + nn);
            nn = r[i].x;
            ans[r[i].i] = gg[r[i].x];
        }
        printf("Case #%d: ", v++);
        for(int i = 0; i < m; i++) {
            printf("%u%c", ans[i], i == m - 1 ? '\n' : ' ');
        }
    }
    return 0;
}