2016vijos 1-3 兔子的晚会(生成函数+倍增FWT)

时间:2021-12-12 17:57:14

2016vijos 1-3 兔子的晚会(生成函数+倍增FWT)

求出序列的生成函数后,倍增FWT

#include<cstdio>

using namespace std;

#define N 2048

const int mod=1e9+;

int inv;

int f[N+];

int Pow(int a,int b)
{
int res=;
for(;b;a=1LL*a*a%mod,b>>=)
if(b&) res=1LL*res*a%mod;
return res;
} void FWT(int *a,int n)
{
int x,y;
for(int d=;d<n;d<<=)
for(int m=d<<,i=;i<n;i+=m)
for(int j=;j<d;++j)
{
x=a[i+j]; y=a[i+j+d];
a[i+j]=x+y; a[i+j+d]=x-y;
a[i+j]-=a[i+j]>=mod ? mod : ;
a[i+j+d]+=a[i+j+d]< ? mod : ;
}
} void IFWT(int *a,int n)
{
int x,y;
for(int d=;d<n;d<<=)
for(int m=d<<,i=;i<n;i+=m)
for(int j=;j<d;++j)
{
x=a[i+j]; y=a[i+j+d];
a[i+j]=1LL*(x+y)*inv%mod; a[i+j+d]=1LL*(x-y+mod)%mod*inv%mod;
}
} int main()
{
//freopen("xor.in","r",stdin);
//freopen("xor.out","w",stdout);
int n,m,L,R;
scanf("%d%d%d%d",&n,&m,&L,&R);
n=*n+;
inv=Pow(,mod-);
int len;
int ans=;
for(int x=L;x<=R;++x)
{
len=;
while(len<=x+m) len<<=;
for(int i=;i<x;++i) f[i]=;
for(int i=x;i<=x+m;++i) f[i]=;
for(int i=x+m+;i<len;++i) f[i]=;
FWT(f,len);
for(int i=;i<len;++i) f[i]=Pow(f[i],n);
IFWT(f,len);
ans+=f[];
ans-=ans>=mod ? mod : ;
}
printf("%d",ans);
}