BZOJ 4589 Hard Nim（FWT+博弈论+快速幂）

【题目链接】 http://www.lydsy.com/JudgeOnline/problem.php?id=4589

【题目大意】

　　有n堆石子，每堆都是m以内的质数，请问后手必胜的局面有几种

【题解】

　　后手必胜，则sg为0，那么就是求n个m以内的数xor为0的情况有几种，
　　首先筛出素数，保存素数的个数数组，利用FWT计算c[i^j]=a[i]*b[j]，
　　计算n次的结果逆向变化回来就是最终的sg个数数组，
　　在计算n次c[i]=a[i]*b[i]的过程中，等价于计算c[i]=a[i]^n，
　　这里我们可以用快速幂优化一个log。

【代码】

#include <cstdio>

#include <algorithm>

using namespace std;

typedef long long LL;

const int N=100000;

const LL mod=1e9+7;

LL a[N],u;

int p[N],n,m;

void FWT(LL*a,int n){

    for(int d=1;d<n;d<<=1)for(int m=d<<1,i=0;i<n;i+=m)for(int j=0;j<d;j++){

        LL x=a[i+j],y=a[i+j+d];

        a[i+j]=(x+y)%mod,a[i+j+d]=(x-y+mod)%mod;

    }

}

void UFWT(LL*a,int n){

    for(int d=1;d<n;d<<=1)for(int m=d<<1,i=0;i<n;i+=m)for(int j=0;j<d;j++){

        LL x=a[i+j],y=a[i+j+d];

        a[i+j]=(x+y)%mod*u%mod,a[i+j+d]=(x-y+mod)%mod*u%mod;

    }

}

LL pow(LL a,LL b,LL p){LL t=1;for(a%=p;b;b>>=1LL,a=a*a%p)if(b&1LL)t=t*a%p;return t;}

int main(){

    for(int i=2;i<=50000;i++)p[i]=1;

    for(int i=2;i<=50000;i++)if(p[i]){

        for(int j=2;i*j<=50000;j++)p[i*j]=0;

    }u=pow(2,mod-2,mod);

    while(~scanf("%d%d",&n,&m)){

        int len=1;while(len<=m)len<<=1;

        for(int i=0;i<len;i++)a[i]=p[i]&(i<=m);

        FWT(a,len);

        for(int i=0;i<len;i++)a[i]=pow(a[i],n,mod);

        UFWT(a,len);

        printf("%lld\n",a[0]);

    }return 0;

}