人生有信仰 数据有梯度 暴力不爆零


bzoj 4589: Hard Nim -- FWT

4589: Hard Nim

Time Limit: 10 Sec  Memory Limit: 128 MB

Description

 
 
Claris和NanoApe在玩石子游戏,他们有n堆石子,规则如下:
1. Claris和NanoApe两个人轮流拿石子,Claris先拿。
2. 每次只能从一堆中取若干个,可将一堆全取走,但不可不取,拿到最后1颗石子的人获胜。
不同的初始局面,决定了最终的获胜者,有些局面下先拿的Claris会赢,其余的局面Claris会负。
Claris很好奇,如果这n堆石子满足每堆石子的初始数量是不超过m的质数,而且他们都会按照最优策略玩游戏,那么NanoApe能获胜的局面有多少种。
由于答案可能很大,你只需要给出答案对10^9+7取模的值。
 
 

 

Input

输入文件包含多组数据,以EOF为结尾。
对于每组数据:
共一行两个正整数n和m。
每组数据有1<=n<=10^9, 2<=m<=50000。
不超过80组数据。
 

 

Output

 

Sample Input

3 7
4 13

Sample Output

6
120

HINT

 

Source

首先nim游戏必胜要所有数异或和为0,直接肯定是不可取的

考虑dp表示前i个数异或和为k的方案数,最后求答案为0的方案数

有一个神奇的位运算多项式快速乘法FWT,然后再加上快速幂就可以解决la

(实际上就是FWT裸题qaq

#include<map>
#include<cmath>
#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define mod 1000000007
#define ll long long
#define N 100010
int pri[N],tot,a[N],ni;
bool vs[N];
void INIT()
{
    for(int i=2;i<N;i++)
    {
        if(!vs[i]) pri[++tot]=i;
        for(int j=1;j<=tot&&pri[j]*i<N;j++)
        {
            vs[i*pri[j]]=1;
            if(i%pri[j]==0) break;
        }
    }
}
int ksm(int a,int b)
{
    int sum=1;
    while(b)
    {
        if(b&1) sum=(ll)sum*a%mod;
        a=(ll)a*a%mod;b>>=1;
    }
    return sum;
}
void FWT(int *x,int n)
{
    int i,j,k,X,Y;
    for(i=1;i<n;i<<=1)
        for(j=0;j<n;j+=i<<1)
            for(k=0;k<i;k++)
            {
                X=x[k+j];Y=x[k+j+i];
                x[k+j]=(X+Y)%mod;
                x[k+j+i]=(X-Y+mod)%mod;
            }
}
void UFWT(int *x,int n)
{
    int i,j,k,X,Y;
    for(i=1;i<n;i<<=1)
        for(j=0;j<n;j+=i<<1)
            for(k=0;k<i;k++)
            {
                X=x[k+j];Y=x[k+j+i];
                x[k+j]=(ll)(X+Y)*ni%mod;
                x[k+j+i]=(ll)(X-Y+mod)*ni%mod;
            }
}
int n,m;
int main()
{
    INIT();
    int tt;
    ni=ksm(2,mod-2);
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(a,0,sizeof(a));
        for(int i=1;i<=tot&&pri[i]<=m;i++) a[pri[i]]=1;
        tt=m;for(m=1;m<=tt;m<<=1);
        FWT(a,m);
        for(int i=0;i<m;i++) a[i]=ksm(a[i],n);
        UFWT(a,m);
        printf("%d\n",a[0]);
    }
    return 0;
}

 

posted @ 2017-12-06 08:24 lkhll 阅读(...) 评论(...) 编辑 收藏