bzoj4589：Hard Nim（fwt+倍增）

某位高一dalao看了一道我不会的题，让我十分好奇，然后听说了一种fwt的算法。
（也许明天出成绩后就退役了…）

题面
经过简单的博弈论分析后的题意：
构造一个n个数的数组，每个都为小于m的质数，使其异或和为0，问方案数。n≤1e9，m≤5e4。

看到n的大小，直接想倍增。再考虑要维护的信息，由于两个数只要相同异或就为0，故要知道异或和为每一个值的方案数。
设前半段和后半段异或和为x的方案数分别为f[x]与g[x]，则

$$ans[x]=\sum_{i Xor j=x}f[i]*g[j]$$
记为 f@g。（读“卷”）
这个东西可以用fwt解决。
跟fft相似，用一个变换TF，使得TF(f)*TF(g)=TF(f@g)。
fwt就是一种可以nlogn做TF与逆TF的神奇的算法。
具体的我还在学习。

由于次数界不会变化，故倍增时可直接用TF形式的向量来快速幂，最后再做逆TF。
（其实对于倍增fft，一开始把次数界开得足够大也可以最后再做逆dft。虽然次数界动态变化，总复杂度也是nlogn，但可以省常数，而且好写。我学fft那么久现在才想到）

#include <iostream>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>

using namespace std;
#define mmst(a, b) memset(a, b, sizeof(a))
#define mmcp(a, b) memcpy(a, b, sizeof(b))

typedef long long LL;

const int N=100100,mo=1e9+7;

int len,n,m;
int a[N],b[N];

int cheng(int a,int b)
{
    int res=1;
    for(;b;b>>=1,a=(LL)a*a%mo)
    if(b&1)
    res=(LL)res*a%mo;
    return res;
}

void fwt(int *a,int ops)
{
    int I2=1;
    if(!ops)
    I2=cheng(2,mo-2);
    for(int k=1;k<len;k<<=1)
    for(int m=k<<1,i=0;i<len;i+=m)
    for(int j=0;j<k;j++)
    {
        int x=(a[i+j]+a[i+j+k])%mo;
        int y=(a[i+j]-a[i+j+k]+mo)%mo;
        a[i+j]=(LL)x*I2%mo;
        a[i+j+k]=(LL)y*I2%mo;
    }
}

int main()
{
    for(int i=2;i<N;i++)
    b[i]=1;

    for(int i=2;i<N;i++)
    if(b[i])
    for(int j=2;j*i<N;j++)
    b[i*j]=0;

    while(~scanf("%d%d",&n,&m))
    {
        mmst(a,0);

        for(int i=1;i<=m;i++)
        a[i]=b[i];

        len=1;
        while(len<=m)
        len<<=1;
        fwt(a,1);

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

        fwt(a,0);
        printf("%d\n",a[0]);
    }

    return 0;
}