[COGS 2259] 异化多肽

题目描述：

QAQ…

题目分析：

很明显我们是可以搞一下指数生成函数的…
然而复杂度是$O(m*n^2)$ 无法优化…
换个方式做生成函数
A(x) 表示 质量为 x 的 氨基酸有几类
$B(x) = \sum_{i=0}^{\infty} A(x)^i$
显然答案是B(x)中的n次项系数
上面的式子即为一个无限等比数列求和
度度的公式$S=a1/(1-q)$
那么 $B(x) = \sum_{i=0}^{\infty} A(x)^i=\frac{1}{1-A(x)}$
很明显B(x)即为(1-A(x))的逆元
我们做一下多项式求逆就好了…

题目链接：

COGS 2259

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;
}