下降幂多项式 lgP6667题解

既然看到了这道“板子”,那还是来写一下题解吧。。。

如果有机会希望能推一下 载谈binominial sum 的做法。

\[\sum_{k=0}^nf(k)\binom n kx^k(1-x)^{n-k} \]

看到组合数和多项式求值就去想下降幂吧,因为没什么别的好办法了。。。

设下降幂多项式 \(g(x)=f(x)\)。

\[\sum_{i=0}^m g_i\sum_{k=0}^n\binom n k k^{\underline i} x^k(1-x)^{n-k} \]

自从联合省选 2020 之后全世界都知道了 \(\binom n m m^{\underline k}=\binom {n-k} {m-k} n^{\underline k}\)。

\[\sum_{i=0}^mg_i\sum_{k=i}^n\binom {n-i} {k-i} n^{\underline i}x^k(1-x)^{n-k} \]

\[\sum_{i=0}^m g_i n^{\underline i} x^i\sum_{k=0}^{n-i}\binom {n-i} kx^k(1-x)^{n-i-k} \]

后面根据二项式定理得到是 \(1\)。

\[\sum_{i=0}^m g_i n^{\underline i} x^i \]

然后把点值乘上 \(e^{-x}\) 就是下降幂多项式了。

#include<cstring>
#include<cstdio>
#include<cctype>
#define IMP(lim,anw) for(i=0;i^(lim);++i)anw
typedef unsigned ui;
const ui M=1e5+5,mod=998244353;
ui buf[M<<2];ui *now=buf,*w[23];
ui n,m,x,f[M],g[M];
inline void swap(ui&a,ui&b){
	ui c=a;a=b;b=c;
}
inline ui Add(const ui&a,const ui&b){
	return a+b>=mod?a+b-mod:a+b;
}
inline ui Del(const ui&a,const ui&b){
	return b>a?a-b+mod:a-b;
}
inline void px(ui*f,ui*g,const ui&len){
	for(ui i=0;i^len;++i)f[i]=1ull*f[i]*g[i]%mod;
}
inline ui pow(ui a,ui b=mod-2){
	ui ans=1;
	for(;b;b>>=1,a=1ull*a*a%mod)if(b&1)ans=1ull*ans*a%mod;
	return ans;
}
inline void NTT_init(const ui&m){
	ui i,j,n,lim(0);while((1<<lim)<m)++lim;n=1<<lim++;
	w[lim]=now;now+=1<<lim-1;
	w[lim][0]=1;w[lim][1]=pow(3,(mod-1>>1)/n);
	for(i=2;i<(1<<lim-1);++i)w[lim][i]=1ull*w[lim][i-1]*w[lim][1]%mod;
	for(j=lim-1;j>=1;--j){
		w[j]=now;now+=1<<j;
		IMP(1<<j,w[j][i]=w[j+1][i<<1]);
	}
}
inline void DFT(ui*f,const ui&n,const ui&M){
	ui i,k,d,x,y,*W,*fl,*fr,len;
	for(len=n>>1,d=M-1;len^0;len>>=1,--d){
		W=w[d];
		for(k=0;k^n;k+=len<<1){
			fl=f+(k);fr=f+(k|len);
			IMP(len,(x=fl[i],y=fr[i])),fl[i]=Add(x,y),fr[i]=1ull*Del(x,y)*W[i]%mod;
		}
	}
}
inline void IDFT(ui*f,const ui&n,const ui&M){
	ui i,k,d,x,y,*W,*fl,*fr,len;
	for(len=1,d=1;len^n;len<<=1,++d){
		W=w[d];
		for(k=0;k^n;k+=len<<1){
			fl=f+(k);fr=f+(k|len);
			IMP(len,(x=fl[i],y=1ull*fr[i]*W[i]%mod)),fl[i]=Add(x,y),fr[i]=Del(x,y);
		}
	}
	k=pow(n);IMP(n,f[i]=1ull*f[i]*k%mod);
	for(i=1;(i<<1)<n;++i)swap(f[i],f[n-i]);
}
signed main(){
	ui i,n,a(1),b(1),ans(0),len(0);
	scanf("%u%u%u",&n,&m,&x);++m;NTT_init(m<<1);g[0]=1;
	for(i=0;i<m;++i)scanf("%u",f+i);while((1<<len)<(m<<1))++len;
	for(i=1;i<m;++i)g[i]=1ull*g[i-1]*i%mod;g[m-1]=pow(g[m-1],mod-2);
	for(i=m-2;i>=1;--i)g[i]=g[i+1]*(i+1ull)%mod;
	for(i=0;i<m;++i)f[i]=1ull*f[i]*g[i]%mod;for(i=1;i<m;i+=2)g[i]=mod-g[i];
	DFT(f,1<<len,len+1);DFT(g,1<<len,len+1);px(f,g,1<<len);IDFT(f,1<<len,len+1);
	for(i=0;i<m;++i)ans=(ans+1ull*f[i]*a%mod*b)%mod,a=1ull*a*(n-i)%mod,b=1ull*b*x%mod;printf("%u",ans);
}
上一篇:多线程的基础应用


下一篇:信息安全遇新挑战 专家热议下一代安全