题目描述:
QAQ…
题目分析:
很明显我们是可以搞一下指数生成函数的…
然而复杂度是
O(m∗n2)
O
(
m
∗
n
2
)
无法优化…
换个方式做生成函数
A(x) 表示 质量为 x 的 氨基酸有几类
B(x)=∑∞i=0A(x)i
B
(
x
)
=
∑
i
=
0
∞
A
(
x
)
i
显然答案是B(x)中的n次项系数
上面的式子即为一个无限等比数列求和
度度的公式
S=a1/(1−q)
S
=
a
1
/
(
1
−
q
)
那么
B(x)=∑∞i=0A(x)i=11−A(x)
B
(
x
)
=
∑
i
=
0
∞
A
(
x
)
i
=
1
1
−
A
(
x
)
很明显B(x)即为(1-A(x))的逆元
我们做一下多项式求逆就好了…
题目链接:
Ac 代码:
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cmath>
#define int long long
const int maxm=2e6+100,mod=1005060097;
int rev[maxm],bin[1<<21];
int A[maxm],B[maxm],C[maxm];
inline int fastpow(int x,int y)
{
int ans=1;
for(;y;y>>=1,x=(x*x)%mod) if(y&1) ans=(ans*x)%mod;
return ans%mod;
}
inline void NTT(int *a,int n,int f)
{
int l=bin[n];
for(int i=0;i<n;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(l-1));
for(int i=0;i<n;i++) if(i<rev[i]) std::swap(a[i],a[rev[i]]);
for(int i=1;i<n;i<<=1)
{
int wn=fastpow(5ll,f>0?(mod-1)/(i<<1):(mod-1)-(mod-1)/(i<<1));
for(int j=0;j<n;j+=(i<<1))
{
int w=1;
for(int k=0;k<i;k++,w=(w*wn)%mod)
{
int x=a[j+k],y=(w*a[i+j+k])%mod;
a[j+k]=(x+y)%mod;
a[i+j+k]=(x-y+mod)%mod;
}
}
}
if(f==-1)
{
int inv=fastpow(n,mod-2);
for(int i=0;i<n;i++) a[i]=(a[i]*inv)%mod;
}
}
void getinv(int *a,int *b,int n)
{
if(n==1)
{
b[0]=fastpow(a[0],mod-2);
return;
}
getinv(a,b,(n+1)>>1);
int m=1;
for(;m<=n;m<<=1);
for(int i=0;i<n;i++) C[i]=a[i];
for(int i=n;i<m;i++) C[i]=0;
NTT(C,m,1),NTT(B,m,1);
for(int i=0;i<m;i++) b[i]=(2*b[i]%mod-C[i]*b[i]%mod*b[i]%mod+mod)%mod;
NTT(B,m,-1);
for(int i=n;i<m;i++) b[i]=0;
}
signed main()
{
freopen("polypeptide.in","r",stdin);
freopen("polypeptide.out","w",stdout);
for(int i=0;i<=20;i++) bin[1<<i]=i;
int n,m;
scanf("%lld%lld",&n,&m);
for(int i=1;i<=m;i++)
{
int x;
scanf("%lld",&x);
if(x<=n) A[x]--;
if(x<=n) if(A[x]<0) A[x]+=mod;
}
A[0]=1;
int l;
for(l=1;l<=2*n;l<<=1);
getinv(A,B,l);
printf("%lld\n",B[n]%mod);
return 0;
}