洛谷 P4389 付公主的背包 题解

题目传送门

题目大意： 有 $n$ 个物品，每个物品的大小为 $v_i$，物品数量无限，现在有个大小为 $s\in [1,m]$ 的背包，问用这些物品装满背包有多少种不同的方案。

题解

容易想到对每个物品构造生成函数，然后乘在一起即可。

但是这题 $n\le 10^5$，$O(n^2\log n)$ 的做法原地去世。

考虑对每个生成函数取 $\ln$，然后加起来再取 $\exp$ 也能得到答案。

考虑第 $i$ 个物品的生成函数，${\sum }_{j=0}{x}^{v_{i}j}=\frac{1}{1-x^{v_i}}$

对 $\frac{1}{1-x^{v_i}}$ 求 $\ln$ 得到：$\ln \frac{1}{1-x^{v_i}}=-\ln (1-x^{v_i})=-\ln (1+(-x^{v_i}))$。

参考这里，可以知道这个东西泰勒展开之后是：$-{\sum }_{j=1}^{\infty }\frac{x^{j\times v_i}}{j}$，和上面的负号搞在一起就是正的了。

所以预处理一下每个 $v_i$ 的贡献，加起来之后求一发 $\exp$ 就是答案了。

代码如下：

#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;
#define maxn 400010
#define bin(x) (1<<(x))
#define mod 998244353
#define MS(F,x) memset(F,0,(4<<(x)))

int n,m,F[maxn],G[maxn];
int ksm(int x,int y){int re=1;for(;(y&1?re=1ll*re*x%mod:0),y;y>>=1,x=1ll*x*x%mod);return re;}
#define INV(x) ksm(x,mod-2)
int inv[maxn],log_2[maxn];
int *w[30];void prep(int N){
	inv[1]=1;for(int i=2;i<=N;i++)inv[i]=1ll*(mod-mod/i)*inv[mod%i]%mod,log_2[i]=ceil(log2(i));
	for(int i=1,wn;i<log_2[N];i++){
		w[i]=new int[bin(i-1)];w[i][0]=1;wn=ksm(3,(mod-1)/bin(i));
		for(int j=1;j<bin(i-1);j++)w[i][j]=1ll*w[i][j-1]*wn%mod;
	}
}
int limit,r[maxn];void work(int lg){for(int i=1;i<bin(lg);i++)r[i]=(r[i>>1]>>1)|((i&1)<<(lg-1));}
void ntt(int *f,int lg,int type=0)
{
	limit=bin(lg);if(type)reverse(f+1,f+limit);
	for(int i=1;i<limit;i++)if(i<r[i])swap(f[i],f[r[i]]);
	for(int mid=1,Lg=1;mid<limit;mid<<=1,Lg++)for(int j=0;j<limit;j+=(mid<<1))for(int i=0;i<mid;i++)
	{int t=1ll*f[j+i+mid]*w[Lg][i]%mod;f[j+i+mid]=(f[j+i]-t+mod)%mod;f[j+i]=(f[j+i]+t)%mod;}
}
void NTT(int *f,int *g,int ln){
	int lg=log_2[ln<<1];work(lg);ntt(f,lg);ntt(g,lg);for(int i=0;i<bin(lg);i++)f[i]=1ll*f[i]*g[i]%mod;
	ntt(f,lg,1);for(int i=0;i<bin(lg);i++)f[i]=1ll*f[i]*inv[limit]%mod;
}
int A[maxn],B[maxn],C[maxn],D[maxn],E[maxn];
void getinv(int *f,int *g,int ln)
{
	if(ln==1){g[0]=INV(f[0]);return;}getinv(f,g,(ln+1)>>1);int lg=log_2[ln<<1];work(lg);
	MS(A,lg);MS(B,lg);memcpy(A,f,ln<<2);memcpy(B,g,ln<<2);
	ntt(A,lg);ntt(B,lg);for(int i=0;i<bin(lg);i++)A[i]=1ll*(2-1ll*A[i]*B[i]%mod+mod)%mod*B[i]%mod;
	ntt(A,lg,1);for(int i=0;i<ln;i++)g[i]=1ll*A[i]*inv[bin(lg)]%mod;
}
void Dao(int *f,int *g,int ln){for(int i=0;i<ln-1;i++)g[i]=1ll*f[i+1]*(i+1)%mod;g[ln-1]=0;}
void Jifen(int *f,int *g,int ln){for(int i=ln-1;i>0;i--)g[i]=1ll*f[i-1]*inv[i]%mod;g[0]=0;}
void getln(int *f,int *g,int ln)
{
	MS(C,log_2[2*ln]);MS(D,log_2[2*ln]);
	getinv(f,C,ln);Dao(f,D,ln);NTT(C,D,ln);Jifen(C,g,ln);
}
void getexp(int *f,int *g,int ln)
{
	if(ln==1){g[0]=1;return;}getexp(f,g,(ln+1)>>1);
	memset(E,0,log_2[ln<<1]);getln(g,E,ln);
	for(int i=0;i<ln;i++)E[i]=(f[i]-E[i]+mod)%mod;E[0]++;NTT(g,E,ln);
}

int main()
{
	scanf("%d %d",&n,&m);prep(m<<2);
	for(int i=1,x;i<=n;i++)scanf("%d",&x),G[x]++;
	for(int i=1;i<=m;i++)if(G[i]){for(int j=1;i*j<=m;j++)
	F[i*j]=(F[i*j]+1ll*G[i]*inv[j]%mod)%mod;G[i]=0;}
	getexp(F,G,m+1);for(int i=1;i<=m;i++)printf("%d\n",G[i]);
}