「TJOI / HEOI2016」求和
题目大意:
求(displaystyle sum_{i=0}^nsum_{j=0}^i egin{Bmatrix}i\ jend{Bmatrix}2^jcdot 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})
(displaystyle F(x+1)=frac{1}{1-2x}=sum_{i=0} (2x)^i)
(displaystyle F'(x+1)=sum_{i=0} 2(i+1) (2x)^i)
(F(x+1)=cfrac{1-2x}{2}F'(x+1))
(displaystyle mathscr F(x+1)=cfrac{1-2x}{2}mathscr F'(x+1)+(n+1)(2x)^n)
(displaystyle mathscr F(x)=cfrac{3-2x}{2}mathscr F'(x)+(n+1)(2x-2)^n)
那么得到
(displaystyle [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}inom{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}inom{n}{k}(-1)^{n-k})
(displaystyle frac{3}{2} [x^{k+1}]mathscr F(x)=[x^k]mathscr F(x)-2^{n}inom{n+1}{k+1}(-1)^{n-k})
最后(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
",ans);
}