loj6268. 分拆数

题意:

1 1 1~ n n n的分拆数。

题解:

考虑其生成函数为 F ( x ) = ∑ i f ( i ) x i F(x)=\sum_if(i)x^i F(x)=if(i)xi
那么 F ( x ) = ∏ i = 1 1 1 − x i F(x)=\prod_{i=1}\frac1{1-x^i} F(x)=i=11xi1
两边取对数得
ln ⁡ F ( x ) = ∑ i = 1 ln ⁡ 1 1 − x i \ln F(x)=\sum_{i=1}\ln\frac1{1-x^i} lnF(x)=i=1ln1xi1

因为
ln ⁡ x = ∑ ( − 1 ) i + 1 ( x − 1 ) i i \ln x=\sum(-1)^{i+1}\frac{(x-1)^i}{i} lnx=(1)i+1i(x1)i

右边泰勒展开得
ln ⁡ F ( x ) = ∑ k = 1 n ∑ i > 1 x k i i \ln F(x)=\sum_{k=1}^n\sum_{i>1}\frac{x^{ki}}{i} lnF(x)=k=1ni>1ixki

那么求出右边后exp回去即可
code:

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#define LL long long
using namespace std;
const LL p=998244353,yg=3;
LL bin[400010],tmp[400010],a[400010],h[400010],g[400010],f[400010];
LL Pow(LL a,LL b)
{
	LL ans=1;
	while(b)
	{
		if(b&1) ans=ans*a%p;
		a=a*a%p;b>>=1;
	}
	return ans;
}
void ntt(LL *a,LL n,LL op)
{
	for(LL i=0;i<n;i++) bin[i]=(bin[i>>1]>>1)|((i&1)*(n>>1));
	for(LL i=0;i<n;i++) if(i<bin[i]) swap(a[i],a[bin[i]]);
	for(LL i=1;i<n;i<<=1)
	{
		LL wn=Pow(yg,op==1?(p-1)/(i<<1):p-1-(p-1)/(i<<1)),t,w;
		for(LL j=0;j<n;j+=i<<1)
		{
			w=1;
			for(LL k=0;k<i;k++)
			{
				t=a[i+j+k]*w%p;w=w*wn%p;
				a[i+j+k]=(a[j+k]-t)%p;a[j+k]=(a[j+k]+t)%p;
			}
		}
	}
	if(op==-1)
	{
		LL Inv=Pow(n,p-2);
		for(LL i=0;i<n;i++) a[i]=a[i]*Inv%p;
	}
}
void pol_inv(LL n,LL *a,LL *g)
{
	if(n==1) {g[0]=Pow(a[0],p-2);return;}
	pol_inv((n+1)>>1,a,g);
	LL N=1;while(N<n<<1) N<<=1;
	copy(a,a+n,tmp);fill(tmp+n,tmp+N,0);
	ntt(tmp,N,1);ntt(g,N,1);
	for(LL i=0;i<N;i++) g[i]=(2-tmp[i]*g[i]%p)*g[i]%p;
	ntt(g,N,-1);fill(g+n,g+N,0);
}
void pol_ln(LL n,LL *a,LL *h)
{
	LL N=1;while(N<n<<1) N<<=1;
	fill(g,g+N,0);pol_inv(n,a,g);
	for(LL i=0;i<n;i++) h[i]=a[i+1]*(i+1)%p;
	fill(h+n,h+N,0);ntt(g,N,1);ntt(h,N,1);
	for(LL i=0;i<N;i++) h[i]=h[i]*g[i]%p;
	ntt(h,N,-1);
	for(LL i=n-1;i>=1;i--) h[i]=h[i-1]*Pow(i,p-2)%p;
	fill(h+n,h+N,0);
}
void pol_exp(LL n,LL *a,LL *f)
{
	if(n==1) {f[0]=1;return;}
	pol_exp(n>>1,a,f);
	fill(h,h+n,0);pol_ln(n,f,h);
	LL N=1;while(N<n<<1) N<<=1;
	tmp[0]=1;for(LL i=1;i<n;i++) tmp[i]=(a[i]-h[i])%p;
	fill(tmp+n,tmp+N,0);
	ntt(tmp,N,1);ntt(f,N,1);
	for(LL i=0;i<N;i++) f[i]=f[i]*tmp[i]%p;
	ntt(f,N,-1);fill(f+n,f+N,0);
}
LL inv[100010],n;
int main()
{
	scanf("%lld",&n);
	inv[0]=inv[1]=1;
	for(LL i=2;i<=n;i++) inv[i]=(p-p/i)*inv[p%i]%p;
	for(LL k=1;k<=n;k++)
		for(LL i=1;i*k<=n;i++) (a[k*i]+=inv[i])%=p;
	int N=1;while(N<=n) N<<=1;
	pol_exp(N,a,f);
	for (LL i=1;i<=n;i++) printf("%lld\n",(f[i]+p)%p);
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值