【BZOJ4555】求和（第二類斯特林數，組合數學，NTT）

題面

BZOJ

題解

推推柿子
\[\sum_{i=0}^n\sum_{j=0}^iS(i,j)·j!·2^j\]
\[=\sum_{i=0}^n\sum_{j=0}^nS(i,j)·j!·2^j\]
\[=\sum_{i=0}^n\sum_{j=0}^nj!·2^j(\frac{1}{j!}\sum_{k=0}^j(-1)^k·C_j^k·(j-k)^i)\]
\[=\sum_{j=0}^n2^j\sum_{k=0}^j(-1)^k·C_j^k·\sum_{i=0}^n(j-k)^i\]
\[=\sum_{j=0}^n2^j\sum_{k=0}^j(-1)^k·\frac{j!}{k!(j-k)!}·\sum_{i=0}^n(j-k)^i\]

#include<iostream>
#include<cstdio>
#include<cstdlib>
 

#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MOD 998244353
#define MAX 500000
const int pr=3;
const int phi=MOD-1;
int fpow(int
 
 a,int b)
{
    int s=1;
    while(b){if(b&1)s=1ll*s*a%MOD;a=1ll*a*a%MOD;b>>=1;}
    return s;
}
int r[MAX],N,M,l;
int jc[MAX],inv[MAX];
int a[MAX],b[MAX],S[MAX];
void NTT(int *P,int opt)
{
    for(int i=0;i<N;++i)if(i<r[i])swap(P[i],P[r[i]]);
    for(int i=1;i<N;i<<=1)
    {
        int W=fpow(pr,phi/(i<<1));
        for(int p=i<<1,j=0;j<N;j+=p)
        {
            int w=1;
            for(int k=0;k<i;++k,w=1ll*w*W%MOD)
            {
                int X=P[j+k],Y=1ll*w*P[i+j+k]%MOD;
                P[j+k]=(X+Y)%MOD;P[i+j+k]=((X-Y)%MOD+MOD)%MOD;
            }
        }
    }
    if(opt==-1)reverse(&P[1],&P[N]);
}
void Work()
{
    M+=N;
    for(N=1;N<=M;N<<=1)++l;
    for(int i=0;i<N;++i)r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
    NTT(a,1);NTT(b,1);
    for(int i=0;i<N;++i)a[i]=1ll*a[i]*b[i]%MOD;
    NTT(a,-1);
    for(int i=0,inv=fpow(N,MOD-2);i<N;++i)a[i]=1ll*a[i]*inv%MOD;
}
int n,ans;
int main()
{
    scanf("%d",&n);
    N=M=n;
    jc[0]=inv[0]=1;
    for(int i=1;i<=n;++i)jc[i]=1ll*jc[i-1]*i%MOD;
    for(int i=1;i<=n;++i)inv[i]=fpow(jc[i],MOD-2);
    for(int i=0;i<=n;++i)a[i]=(i&1)?-1:1;
    for(int i=0;i<=n;++i)a[i]=(1ll*a[i]*inv[i]%MOD+MOD)%MOD;
    for(int i=2;i<=n;++i)b[i]=(fpow(i,n+1)-1+MOD)%MOD;
    for(int i=2;i<=n;++i)b[i]=1ll*b[i]*inv[i]%MOD;
    for(int i=2;i<=n;++i)b[i]=1ll*b[i]*fpow(i-1,MOD-2)%MOD;
    b[0]=1;b[1]=n+1;
    Work();
    for(int i=0,p=1;i<=n;++i,p=(p+p)%MOD)ans=(ans+1ll*jc[i]*p%MOD*a[i]%MOD)%MOD;
    printf("%d\n",ans);
    return 0;
}

【BZOJ4555】求和（第二類斯特林數，組合數學，NTT）