【BZOJ4589】Hard Nim(FWT)

【BZOJ4589】Hard Nim(FWT)

题面

BZOJ

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组数据。

Sample Input

3 7

4 13

Sample Output

6

120

题解

相当于变相的询问,对于给定的所有小于\(m\)的质数中
选出\(n\)个数,他们的异或和为\(0\)的方案数
如果\(n\)很小,我们可以一次次的做\(FWT\)
但是\(n<=10^9\),所以用快速幂优化一下就行了
中间没有必要做\(IFWT\)转换回去
一直用\(FWT\)做最后再做一次\(IFWT\)就行了

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MOD (1000000007)
#define inv2 (500000004)
inline int read()
{
    RG int x=0,t=1;RG char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
int a[1<<16],b[1<<16],N;
void FWT(int *P,int opt)
{
	for(int i=2;i<=N;i<<=1)
		for(int p=i>>1,j=0;j<N;j+=i)
			for(int k=j;k<j+p;++k)
			{
				int x=P[k],y=P[k+p];
				P[k]=(x+y)%MOD;P[k+p]=(x-y+MOD)%MOD;
				if(opt==-1)P[k]=1ll*P[k]*inv2%MOD,P[k+p]=1ll*P[k+p]*inv2%MOD;
			}
}
int n,m;
bool zs[55555];
void prime()
{
	zs[1]=true;
	for(int i=2;i<=50000;++i)
		if(!zs[i])
			for(int j=i+i;j<=50000;j+=i)
				zs[j]=true;
}
void fpow(int *a,int *b,int p)
{
	FWT(a,1);FWT(b,1);
	while(p)
	{
		if(p&1)for(int i=0;i<N;++i)b[i]=1ll*b[i]*a[i]%MOD;
		for(int i=0;i<N;++i)a[i]=1ll*a[i]*a[i]%MOD;
		p>>=1;
	}
	FWT(b,-1);
}
int main()
{
	prime();
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		for(N=1;N<=m;N<<=1);
		memset(a,0,sizeof(a));memset(b,0,sizeof(b));
		for(int i=1;i<=m;++i)
			if(!zs[i])a[i]=b[i]=1;
		fpow(a,b,n-1);
		printf("%d\n",b[0]);
	}
	return 0;
}

posted @ 2018-05-21 09:20  小蒟蒻yyb  阅读(693)  评论(0编辑  收藏  举报