「TJOI / HEOI2016」求和

题目大意：

求 \(\displaystyle \sum_{i=0}^n\sum_{j=0}^i \begin{Bmatrix}i\\ j\end{Bmatrix}2^j\cdot j!\) 。

由于第二类斯特林数的生成函数 \(S_m(x)=\cfrac{1}{m!}(e^x-1)^m\) 。

所以求的东西就是 \(\displaystyle F(x)=\sum_{i=0} (2e^x-2)^i=\frac{1}{3-2e^x}\) 前 \(n\) 项系数。

可以暴力求逆

线性解法：思路

要求 \(\displaystyle \frac{1}{3-2e^x}\) 的前 \(n\) 项 \([x^i]\) 乘 \(i!\) 的和。

设 \(\displaystyle G(x)=e^x,F(x)=\frac{1}{3-2x}\) 。

那么我们需要求得 \(\mathscr F(x+1)=F(x+1) \mod x^{n+1}\) 。

\[ \begin{aligned} F(x+1)&=\frac{1}{1-2x}=\sum_{i=0} (2x)^i \\ F'(x+1)&=\sum_{i=0} 2(i+1) (2x)^i \\ F(x+1)&=\cfrac{1-2x}{2}F'(x+1) \\ \mathscr F(x+1)&=\cfrac{1-2x}{2}\mathscr F'(x+1)+(n+1)(2x)^n \\ \mathscr F(x)&=\cfrac{3-2x}{2}\mathscr F'(x)+(n+1)(2x-2)^n \end{aligned} \]

那么得到：

\[ \begin{aligned} [x^k]\mathscr F(x)&=\frac{3}{2}(k+1)[x^{k+1}]\mathscr F(x)-k[x^k]\mathscr F(x)+(n+1)2^{n}\binom{n}{k}(-1)^{n-k} \\ \displaystyle \frac{3}{2}(k+1)[x^{k+1}]\mathscr F(x)&=(k+1)[x^k]\mathscr F(x)-(n+1)2^{n}\binom{n}{k}(-1)^{n-k} \\ \displaystyle \frac{3}{2} [x^{k+1}]\mathscr F(x)&=[x^k]\mathscr F(x)-2^{n}\binom{n+1}{k+1}(-1)^{n-k} \end{aligned} \]

最后得到： \(\displaystyle \sum_{i=0}^n [x^i]F(G(x))=\sum_{i=0}^n [x^i]\mathscr F(G(x))=\sum \mathscr F_i \sum_{j=0}^n j! [x^j]G^k(x)\) 。

\(\displaystyle [x^0]\mathscr F(x)=[x^0]\sum_{i=0}^n(2x-2)^i=\sum_{i=0}^n (-2)^i=\frac{1-(-2)^{n+1}}{3}\)

\(\sum_{j=0}^n j! [x^j]G^k(x)\) 就是一个等比数列求和，可以用线性筛求 \(i^k\) 即可做到线性求解。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define rep(i,a,b) for(int i=a,i##end=b;i<=i##end;++i)
#define drep(i,a,b) for(int i=a,i##end=b;i>=i##end;--i)

const int N=1e5+10,P=998244353;

int n;
int I[N],J[N],Inv[N],Pow[N],q[N],F[N];
ll qpow(ll x,ll k=P-2){
	ll res=1;
	for(;k;k>>=1,x=x*x%P) if(k&1) res=res*x%P;
	return res;
}

void Sieve_Pow(int n){
	static int notpri[N],pri[N],pc;
	rep(i,2,n) {
		if(!notpri[i]) pri[++pc]=i,Pow[i]=qpow(i,n);
		for(int j=1;j<=pc && 1ll*i*pri[j]<=n;++j) {
			notpri[i*pri[j]]=1,Pow[i*pri[j]]=1ll*Pow[i]*Pow[pri[j]]%P;
			if(i%pri[j]==0) break;
		}
	}
}
int C(int n,int m){ return 1ll*J[n]*I[m]%P*I[n-m]%P; }

int main(){
	scanf("%d",&n),Inv[0]=Inv[1]=1;
	rep(i,2,n+1) Inv[i]=1ll*(P-P/i)*Inv[P%i]%P;
	rep(i,*I=*J=1,n+1) I[i]=1ll*I[i-1]*Inv[i]%P,J[i]=1ll*J[i-1]*i%P;
	Sieve_Pow(n+1);
	ll p=qpow(2,n);
	q[0]=1,q[1]=n+1;
	rep(i,2,n) q[i]=1ll*(Pow[i]-1)*Inv[i-1]%P;
	F[0]=(((n&1)?P-p*2%P:p*2%P)+1)*(P+1)/3%P;
	rep(i,0,n-1) {
		int t=p%P*C(n+1,i+1)%P;
		if((n-i+1)&1) t=P-t;
		F[i+1]=(F[i]+t)*2ll%P*(P+1)/3%P;
	}
	int ans=0;
	rep(i,0,n) ans=(ans+1ll*F[i]*q[i])%P;
	printf("%d\n",ans);
}