bzoj 4589 Hard Nim —— FWT

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=4589

先手必败，是一开始所有石子的异或和为0；

生成函数 (x^pri[1] + x^pri[2] + ... + x^pri[k])ⁿ，pri[k] <= m

FWT求解即可；

而且不要快速幂里面每次变换来变换去的，只有快速幂前后需要变换。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
int const xn=(<<),mod=1e9+;
int n,m,a[xn],b[xn],lim,inv,cnt,pri[xn];
bool vis[xn];
void init()
{
  int mx=xn-;
  for(int i=;i<=mx;i++)
    {
      if(!vis[i])pri[++cnt]=i;
      for(int j=;j<=cnt&&(ll)i*pri[j]<=mx;j++)
    {
      vis[i*pri[j]]=;
      if(i%pri[j]==)break;
    }
    }
}
int upt(int x){while(x>=mod)x-=mod; while(x<)x+=mod; return x;}
ll pw(ll a,int b)
{
  ll ret=;
  for(;b;b>>=,a=(a*a)%mod)if(b&)ret=(ret*a)%mod;
  return ret;
}
void fwt(int *a,int tp)
{
  for(int mid=;mid<lim;mid<<=)
    for(int j=,len=(mid<<);j<lim;j+=len)
      for(int k=;k<mid;k++)
    {
      int x=a[j+k],y=a[j+mid+k];
      a[j+k]=upt(x+y); a[j+mid+k]=upt(x-y);
      if(tp==-)a[j+k]=(ll)a[j+k]*inv%mod,a[j+mid+k]=(ll)a[j+mid+k]*inv%mod;
    }
}
int main()
{
  inv=pw(,mod-); init();
  while(scanf("%d%d",&n,&m)==)
    {
      memset(a,,sizeof a);
      for(int i=;i<=m;i++)if(!vis[i])a[i]=;
      memset(b,,sizeof b); b[]=;
      lim=; while(lim<=m)lim<<=;
      fwt(a,); fwt(b,);
      for(;n;n>>=)
    {
      if(n&)for(int i=;i<lim;i++)b[i]=(ll)a[i]*b[i]%mod;
      for(int i=;i<lim;i++)a[i]=(ll)a[i]*a[i]%mod;
    }
      fwt(b,-);
      printf("%d\n",b[]);
    }
  return ;
}