洛谷 P4389 付公主的背包 多项式exp

题目描述
这个背包最多可以装$10^5$大小的东西
付公主有$n$种商品，她要准备出摊了
每种商品体积为$V_i$，都有$10^5$件
给定$m$，对于$s\in [1,m]$，请你回答用这些商品恰好装$s$体积的方案数

输入输出格式

输入格式：
第一行$n,m$

第二行$V_1$,$V_n$

输出格式：
m行，第i行代表s=i时方案数，对998244353取模

输入输出样例

输入样例#1：
2 4
1 2
输出样例#1：
1
2
2
3
说明

对于30%的数据，$n<=3000,m<=3000$

对于60%的数据，纯随机生成

对于100%的数据， $n<=100000,m<=100000$

对于100%的数据，$V_i<=m$

分析：
对于一种体积为$v$的物品可以用多项式${\sum }_{i\ge 0}{x}^{vi}$代替。
所有的多项式的卷积就是答案。这样才能保证没有重复。
直接卷积肯定不行，考虑先对所有多项式求ln，然后相加，最后多项式exp。
然后发现求ln后的多项式为${\sum }_{i>0}\frac{1}{i}{x}^{vi}$。
多项式exp即可。

代码：

// luogu-judger-enable-o2
#include <iostream>
#include <cstdio>
#include <cmath>
#define LL long long

const LL mod=998244353;
const LL G=3;
const int maxn=6e5+7;

using namespace std;

int n,m,v,len;
int r[maxn];
LL a[maxn],b[maxn],c[maxn],s[maxn],f[maxn],g[maxn],h[maxn],num[maxn],ny[maxn],inv[maxn],w[maxn];

LL ksm(LL x,LL y)
{
    if (y==1) return x;
    LL c=ksm(x,y/2);
    c=(c*c)%mod;
    if (y&1) c=(c*x)%mod;
    return c;
}

void ntt(LL *a,int f)
{
    for (int i=0;i<len;i++)
    {
        if (i<r[i]) swap(a[i],a[r[i]]);
    }
    w[0]=1;
    for (int i=2;i<=len;i*=2)
    {
        LL wn;
        if (f==1) wn=ksm(G,(LL)(mod-1)/i);
             else wn=ksm(G,(LL)(mod-1)-(mod-1)/i);
        for (int j=i/2;j>=0;j-=2) w[j]=w[j/2];
        for (int j=1;j<i/2;j+=2) w[j]=(w[j-1]*wn)%mod;
        for (int j=0;j<len;j+=i)
        {
            for (int k=0;k<i/2;k++)
            {
                LL u=a[j+k],v=a[j+k+i/2]*w[k]%mod;
                a[j+k]=(u+v)%mod;
                a[j+k+i/2]=(u+mod-v)%mod;
            }
        }
    }
    if (f==-1)
    {
        LL inv=ksm(len,mod-2);
        for (int i=0;i<len;i++) a[i]=a[i]*inv%mod;
    }
}

void NTT(LL *x,LL *y,LL *z,int n,int m)
{
    len=1;
    int k=0;
    while (len<=(n+m)) len*=2,k++;
    for (int i=0;i<len;i++)
    {
        r[i]=(r[i>>1]>>1)|((i&1)<<(k-1));
    }
    for (int i=0;i<len;i++)
    {
        if (i<n) a[i]=x[i]; else a[i]=0;
        if (i<m) b[i]=y[i]; else b[i]=0;
    }
    ntt(a,1); ntt(b,1);
    for (int i=0;i<len;i++) z[i]=(a[i]*b[i])%mod;
    ntt(z,-1);
}

void getinv(LL *f,LL *g,int deg)
{
    if (deg==1)
    {
        g[0]=ksm(f[0],mod-2);
        return;
    }
    int d=(deg+1)/2;
    getinv(f,g,d);
    NTT(f,g,c,deg,d);
    c[0]=(2+mod-c[0])%mod;
    for (int i=1;i<deg;i++) c[i]=(mod-c[i])%mod;
    NTT(c,g,g,deg,d);
    for (int i=deg;i<len;i++) g[i]=0;
}

void tran(LL *a,LL *b,int deg)
{
    for (int i=1;i<deg;i++) 
      b[i-1]=(i*a[i])%mod;
    b[deg-1]=0;
}

void intran(LL *a,LL *b,int deg)
{
    for (int i=1;i<deg;i++) b[i]=a[i-1]*ksm(i,mod-2)%mod;
    b[0]=0;
}

void ln(LL *f,LL *g,int n)
{
    getinv(f,inv,n);
    tran(f,f,n);
    NTT(f,inv,f,n,n);
    intran(f,g,n);
}

void solve(LL *f,LL *g,int deg)
{
    if (deg==1)
    {
        g[0]=1;				
        return;
    }
    int mid=(deg+1)/2;
    solve(f,g,mid);
    for (int i=0;i<mid;i++) s[i]=g[i];
    ln(s,h,mid);
    h[0]=(f[0]+1+mod-h[0])%mod;
    for (int i=1;i<mid;i++) h[i]=(f[i]+mod-h[i])%mod;
    NTT(h,g,g,mid,mid);
    for (int i=deg;i<len;i++) g[i]=h[i]=0;
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&v);
        num[v]++;
    }	
    for (int i=1;i<=m;i++) ny[i]=ksm(i,mod-2);
    for (int i=1;i<=m;i++)
    {
        if (num[i])
        {
            for (int j=1;j*i<=m;j++)
            {
                f[i*j]=(f[i*j]+num[i]*ny[j]%mod)%mod;
            }
        }
    }
    m++;	
    solve(f,g,m<<1);
    for (int i=1;i<m;i++) printf("%lld\n",g[i]);
}