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

Description

$Claris$和$NanoApe$在玩石子游戏，他们有$n$堆石子，规则如下：

$1$. $Claris$和$NanoApe$两个人轮流拿石子，$Claris$先拿。

$2$.每次只能从一堆中取若干个，可将一堆全取走，但不可不取，拿到最后$1$颗石子的人获胜。

不同的初始局面，决定了最终的获胜者，有些局面下先拿的$Claris$会赢，其余的局面$Claris$会负。

$Claris$很好奇，如果这$n$堆石子满足每堆石子的初始数量是不超过$m$的质数，而且他们都会按照最优策略玩游戏，那么$NanoApe$能获胜的局面有多少种。

由于答案可能很大，你只需要给出答案对$10^9+7$取模的值。

Input

输入文件包含多组数据，以$EOF$为结尾。

对于每组数据：

共一行两个正整数$n$和$m$。

每组数据有$1\le n\le 10^9, 2\le m\le 50000$。

不超过$80$组数据。

Output

对于每组用例，输出使得$NanoApe$能获胜的局面种数，结果对$10^9+7$取模

Sample Input

3 7

4 13

Sample Output

6

Solution

先手败即选出$n$个不超过$m$的素数使其异或和为$0$，$FWT$后快速幂然后$UFWT$即可

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define maxn 50005
#define maxfwt 65536+5
const int mod=1e9+7;
const int inv2=(mod+1)/2;
void FWT(int *a,int n)  
{  
    for(int d=1;d<n;d<<=1)  
        for(int i=0;i<n;i+=(d<<1))  
            for(int j=0;j<d;j++)  
                {  
                    int 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(int *a,int n)  
{  
    for(int d=1;d<n;d<<=1)  
        for(int i=0;i<n;i+=(d<<1))  
            for(int j=0;j<d;j++)  
            {  
                int x=a[i+j],y=a[i+j+d];  
                a[i+j]=(ll)(x+y)*inv2%mod,a[i+j+d]=(ll)(x-y+mod)*inv2/mod;  
            }  
} 
int Pow(int a,int b)
{
    int ans=1;
    while(b)
    {
        if(b&1)ans=(ll)ans*a%mod;
        a=(ll)a*a%mod;
        b>>=1;
    }
    return ans;
}
int res,p[maxn],mark[maxn];
void get_prime(int n=50000)
{
    res=0;
    for(int i=2;i<=n;i++)
        if(!mark[i])
        {
            p[res++]=i;
            for(int j=2*i;j<=n;j+=i)mark[j]=1;
        }
}
int n,m,a[maxfwt];
int main()
{
    get_prime();
    while(~scanf("%d%d",&n,&m))
    {
        memset(a,0,sizeof(a));
        for(int i=0;i<res&&p[i]<=m;i++)a[p[i]]=1;
        int len=1;
        while(len<=m)len<<=1;
        FWT(a,len);
        for(int i=0;i<len;i++)
            if(a[i])a[i]=Pow(a[i],n);
        UFWT(a,len);
        printf("%d\n",a[0]);
    }
    return 0;
}