Nim游戏的规则如下:A B两个人轮流拿石子,A先拿。每次只能从一堆中取若干个,可将一堆全取走,但不可不取,拿到最后1颗石子的人获胜。不同的初始局面,决定了最终的获胜者,有些局面下先拿的会赢,其余的先手会负。
共有N堆石子,每堆石子的数量都是 <= M的质数。求符合条件的先手负的局面有多少种。例如: N = 3(共3堆石子),M = 7,小于M的质数为:2, 3, 5, 7。如果初始局面为(2,5,7)则先手负。同理,(5,7,2)以及其它(2,5,7)的排列,都符合先手负的条件,并且除此之外的其他组合都不能做到先手负。所以符合条件的局面共有3! = 6种。由于符合条件的局面很多,输出Mod 10^9 + 7的结果即可。
注:2 2 3 3同 2 3 2 3 算2种不同的局面。
Input
输入共2个数:N, M中间用空格分隔。N为石子的堆数,M为每堆石子数量的上限,并且每堆石子的数量为质数。(2 <= N <= 10^9,2 <= M <= 50000)。
Output
输出符合条件的局面的数量Mod 10^9 + 7。
Input示例
10 100
Output示例
294844622
FWT~
——快速沃尔什变换。
听说是裸题,根本没看出来呢。哼。
如果想抄板子的话,可以看看这个:http://blog.csdn.net/liangzhaoyang1/article/details/52819835~
a数组要开成二倍~
#include<cstdio>
#include<iostream>
using namespace std;
#define ll long long
#define N 50001
const int mod=1e9+7;
int n,m,l,kkz,a[N<<1],c[N];
bool b[N];
int mi(int u,int v)
{
int now=1;
for(;v;v>>=1,u=(ll)u*u%mod) if(v&1) now=(ll)now*u%mod;
return now;
}
void fwtf(int l,int r)
{
if(l==r) return;
int mid=l+r>>1;
for(int i=l,j=mid+1;i<=mid;i++,j++)
{
int x=(ll)(a[j]-a[i]+mod)*kkz%mod;
a[i]=(ll)(a[i]+a[j])*kkz%mod;
a[j]=x;
}
fwtf(l,mid);fwtf(mid+1,r);
}
void fwt(int l,int r)
{
if(l==r) return;
int mid=l+r>>1;
fwt(l,mid);fwt(mid+1,r);
for(int i=l,j=mid+1;i<=mid;i++,j++)
{
int x=(a[i]+a[j])%mod;
a[i]=(a[i]-a[j]+mod)%mod;
a[j]=x;
}
}
int main()
{
for(int i=2;i<N;i++)
{
if(!b[i]) c[++c[0]]=i;
for(int j=1;j<=c[0] && (ll)i*c[j]<N;j++)
{
b[i*c[j]]=1;
if(!(i%c[j])) break;
}
}
kkz=mi(2,mod-2);
scanf("%d%d",&n,&m);
for(int i=1;i<=c[0] && c[i]<=m;i++) a[c[i]]=1;
for(l=1;l<=m;l<<=1);l--;
fwt(0,l);
for(int i=0;i<=l;i++) a[i]=mi(a[i],n);
fwtf(0,l);
printf("%d\n",a[0]);
return 0;
}