hdu6040 Hints of sd0061（寻找第b[i]+1小的数）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6040

题意：
用给定函数生成n个unsigned类型的数，用a数组保存，询问m次，每次询问a数组中第b[i]+1小的数是什么。

思路：
对a数组排序是不行的。题目中case<=10, n<1e7, m<=100, O(case * m *nlog(n))的复杂度是会超时的。

使用nth_element()函数。题目中有一点，b[i]+b[j] <= b[k], b[i]!=b[j], b[i] < b[k] , b[j] < b[k]，我们把b[i]，b[j]，b[k]看成a数组中的下标ii, jj, kk, 那么kk可能比ii和jj大很多。那么，我们先把b数组排序，先查询a[kk], 再查询a[ii]和a[jj]，这样我们可以把n的规模减小。

排序：
对b数组下标进行排序。另开一个数组idx，记录每个b[i]中的i，重定义小于号，将数组idx中下标i满足b[i]>b[j]的idx[i]排到前面。

代码：

#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#include <cstdio>

using namespace std;

const int maxn = 1e7+10;
unsigned x, y, z;
int n, m, A, B, C, Case, b[110], idx[110];
unsigned a[maxn], ans[110];

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;
}

bool cmp(int i, int j){
    return b[i] > b[j];
}

int main(){
    Case = 0;
    while(cin>>n>>m>>A>>B>>C){
        x=A; y=B; z=C;        
        for(int i=0; i<m; ++i){
            cin>>b[i];
            idx[i] = i;
        }
        for(int i=0; i<n; ++i){
            a[i] = rng61();
        }
        sort(idx, idx+m, cmp);
        int last = n;
        for(int i=0; i<m; ++i){
            int pos = b[idx[i]];
            nth_element(a, a+pos, a+last);
            ans[idx[i]] = a[pos];
            last = pos;
        }
        cout<<"Case #"<<++Case<<":";
        for(int i=0; i<m; ++i){
            cout<<" "<<ans[i];
        }
        cout<<endl;
    }
    return 0;
}