20200501 zr T1【k次多项式的n次幂的前x项系数】

题目描述：

题目分析：

第一种是链式法则，第二种是$(uv{)}^{\prime }=u^{\prime }v+u{v}^{\prime }$
令$f_i=P^n(x)[i]$，$g_i=P(x)[i]$，已知$g$且最高项次数为$k$，我们要求$f$的前$X$项
首先将第二个式子的第二项移到左边，可以得到：
$$n{P}^n(x)P^{\prime }(x)=(P^n(x){)}^{\prime }P(x)$$
即
$$n\sum _{i=0}^t{f}_i\ast (t-i+1)\ast g_{t-i+1}=\sum _{i=0}^t(i+1)\ast f_{i+1}\ast g_{t-i}$$
可以发现求导之后右边出现了$f_{t+1}$这一项，我们把它提出来，并按照$g$的顺序书写右边：
$$\begin{gathered} f_{t+1}\ast (t+1)\ast g_0=\sum _{i=1}^{k}n\ast g_i\ast i\ast f_{t-i+1}-g_i\ast (t-i+1)\ast f_{t-i+1} \\ =\sum _{i=1}^k{g}_i\ast f_{t-i+1}\ast ((n+1)i-t-1) \end{gathered}$$
这样我们就可以用$O(k)$的时间求出下一项，总复杂度$O(kX)$，不难发现这是求k次多项式的n次幂的前x项系数的通用方法。

Code：

#include<bits/stdc++.h>
#define maxn 10000005
using namespace std;
const int mod = 998244353;
int n,K,X,ans,sum,f[maxn],p[maxn],inv[maxn];
inline int Pow(int a,int b){
	int s=1; for(;b;b>>=1,a=1ll*a*a%mod) if(b&1) s=1ll*s*a%mod;
	return s;
}
int main()
{
	scanf("%d%d%d",&n,&K,&X);
	for(int i=0;i<=K;i++) scanf("%d",&p[i]),sum+=p[i];
	for(int i=0,Iv=Pow(sum,mod-2);i<=K;i++) p[i]=1ll*p[i]*Iv%mod;
	inv[0]=inv[1]=1; int Iv0=Pow(p[0],mod-2);
	sum=f[0]=Pow(p[0],n);
	for(int i=1;i<X;i++){
		if(i>1) inv[i]=1ll*(mod-mod/i)*inv[mod%i]%mod;
		for(int j=1;j<=K&&j<=i;j++) f[i]=(f[i]+(1ll*(n+1)*j-i)%mod*p[j]%mod*f[i-j])%mod;
		f[i]=1ll*f[i]*inv[i]%mod*Iv0%mod;
		ans=(ans+1ll*f[i]*i)%mod;
		sum=(sum+f[i])%mod;
	}
	printf("%d\n",((ans+(1ll-sum)*X)%mod+mod)%mod);
}