noip模拟赛密码

题目描述

YJC把核弹发射密码忘掉了……其实是密码被加密了，但是YJC不会解密。密码由n个数字组成，第i个数字被加密成了如下形式：第k小的满足(2^L)|(P-1)且P为质数的P。YJC希望你能帮他算出密码是多少。

输入输出格式

输入格式：

第一行包含一个整数n，表示密码中的数字个数。

接下来n行每行两个整数L和k，表示一个数字的加密形式。

注意，输入格式变更，请注意L和k的先后顺序

输出格式：

输出n行，第i行一个整数，表示第i个数字。

输入输出样例

输入样例#1：

输出样例#1：

1998585857

998244353

说明

对于50%的数据，满足18≤n,L≤1000。

对于100%的数据，满足12≤n,L≤500000，保证答案<2^31。

分析：比较考验人品的一道题。

暴力找的话有30分.时间都浪费在了怎么判断素数上，而且询问很多，每个数可能会被判断多次，考虑怎么优化它们.

对于素数的判断，可以用一个比较考验人品的算法miller rabin测试，每一次都有大约1/4的概率会出错，因此要多判几次，但是多判几次又会超时，将次数控制在3次到5次就可以了.这里利用费马小定理+快速幂进行判断.

接下来考虑怎么不重复判断同一个数。我们将询问的L按照降序排序,符合条件的数肯定是k*2^i+1的形式，我们先从大到小枚举i,再枚举k,k枚举的是奇数，因为如果我们一个一个地枚举，如果k=2,那么就会跳到2^(i+1) + 1上去，而我们之前已经计算过了这个数，会造成重复.将所得的素数排个序.

最后考虑怎么快速得到第k小的p,因为我们已经将所得的素数排好序了，所以我们可以二分查找，由于之前我们将询问的L按照降序排列，那么得到的满足要求的数就构成了一个又一个的区间，那么利用树状数组来维护有多少个比它小的就可以了.

听说还有还可以用等差数列筛法，orz

#include <cstdio>

#include <cstring>

#include <iostream>

#include <algorithm>

using namespace std;

const int maxn =  << ,maxm = ;

long long n,cnt,num[maxm],prime[maxm],t[maxm],ans[maxm],c[maxm];

struct node

{

    long long l, k;

    int id;

}e[maxm];

long long qpow(long long a, long long b, long long mod)

{

    long long ans = ;

    while (b)

    {

        if (b & )

            ans = (ans * a) % mod;

        b >>= ;

        a = (a * a) % mod;

    }

    return ans;

}

bool isok(long long x)

{

    if (qpow(, x - , x) ==  && qpow(, x - , x) ==  && qpow(, x - , x) == )

        return true;

    return false;

}

bool cmp1(node a, node b)

{

    return a.l > b.l;

}

int main()

{

    scanf("%lld", &n);

    for (int i = ; i <= n; i++)

    {

        scanf("%lld%lld", &e[i].l, &e[i].k);

        e[i].id = i;

    }

    sort(e + , e +  + n, cmp1);

    for (int i = ; i >= ; num[i--] = cnt)

        for (int j = ; 1LL * j * ( << i) < (1LL << ) - ; j += )

            if (isok(j * ( << i) + ))

        prime[++cnt] = j * ( << i) + ;

    memcpy(t, prime, sizeof(prime));

    sort(t + , t +  + cnt);

    for (int i = ; i <= cnt; i++)

        prime[i] = lower_bound(t + , t +  + cnt, prime[i]) - t; //实际上是把符合要求的质数和离散化了

    int cur = ;

    for (int i = ; i <= n; i++)

    {

        while (cur <= num[e[i].l])

        {

            for (int j = prime[cur]; j < maxn; j += j & -j)

                c[j]++;

            cur++;

        }

        int l = , r = maxn, sum = ;;

        while (l<r)

        {

            int mid = (l + r) >> ;

            sum += c[mid];

            if (sum >= e[i].k)

            {

                sum -= c[mid];

                r = mid;

            }

            else l = mid + ;

        }

        ans[e[i].id] = t[l];

    }

    for (int i = ; i <= n; i++)

        printf("%lld\n", ans[i]);

    return ;

}