BZOJ4589 Hard Nim（快速沃尔什变换模板）

终于抽出时间来学了学，比FFT不知道好写到哪里去。

#include <cstdio>

typedef long long ll;
const int N=,p=1e9+;
int k,m,n,a[N],pi[N];
bool pr(int x) {for(int i=;i*i<=x;i++) if(x%i==) return ; return ;}
ll pw(ll a,int b) {ll r=; for(;b;b>>=,a=a*a%p) if(b&) r=r*a%p; return r;}

void fwt(int *a,ll f) {
    for(int i=,x,y;i<n;i<<=) for(int j=;j<n;j+=i<<) for(int k=;k<i;k++)
        x=a[j+k],y=a[j+k+i],a[j+k]=(x+y)*f%p,a[j+k+i]=(x-y+p)*f%p;
}

int main() {
    for(int i=;i<;i++) if(pr(i)) pi[i]=;
    while(~scanf("%d%d",&k,&m)) {
        for(n=;n<=m;n<<=);
        for(int i=;i<=m;i++) a[i]=pi[i];
        for(int i=m+;i<n;i++) a[i]=;
        fwt(a,);
        for(int i=;i<n;i++) a[i]=pw(a[i],k);
        fwt(a,),printf("%d\n",a[]);
    }
    return ;
}