[bzoj4555][TJOI&HEOI2016]求和

题目大意

这里写图片描述

向卷积形式出发

ans=i=0nj=0iS(i,j)2jj!

注意到第二类斯特林数的意义,将i个数分成j个集合的方案数,也就是说,当 i<j 时,结果会为0,所以j的枚举上界可以直接改为n。
ans=i=0nj=0nS(i,j)2jj!

代入第二类斯特林数计算公式
ans=i=0nj=0n2jk=0j(1)kCkj(jk)i

拆一下组合数
ans=i=0nj=0nj!2jk=0j(1)kk!(jk)i(jk)!

可以把第一个sigma移到最后
ans=j=0nj!2jk=0j(1)kk!ni=0(jk)i(jk)!

好了!我们设 a[i]=(1)ii! b[i]=nk=0iki!
把后面整个sigma看做c[j],那么有c=a*b。
所以上NTT咯。

#include<cstdio>
#include<cmath>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
typedef long long ll;
typedef double db;
const int maxn=100000+10,mo=998244353,GG=3;
int a[maxn*4],b[maxn*4],c[maxn*4],w[maxn*4],tt[maxn*4],f[maxn],g[maxn];
int A[maxn*4],B[maxn*4];
int i,j,k,l,t,n,m,len,ni,ans;
db ce;
int quicksortmi(int x,int y){
    if (!y) return 1;
    int t=quicksortmi(x,y/2);
    t=(ll)t*t%mo;
    if (y%2) t=(ll)t*x%mo;
    return t;
}
void DFT(int *a,int sig){
    int i;
    fo(i,0,len-1){
        int p=0;
        for(int j=0,tp=i;j<ce;j++,tp/=2) p=(p<<1)+(tp%2);
        tt[p]=a[i];
    }
    for (int m=2;m<=len;m*=2){
        int half=m/2,bei=len/m;
        fo(i,0,half-1){
            int wi=sig>0?w[i*bei]:w[len-i*bei];
            for(int j=i;j<len;j+=m){
                int u=tt[j],v=(ll)tt[j+half]*wi%mo;
                tt[j]=(u+v)%mo;
                tt[j+half]=(u-v)%mo;
            }
        }
    }
    if (sig==-1)
        fo(i,0,len-1) tt[i]=(ll)tt[i]*ni%mo;
    fo(i,0,len-1) a[i]=tt[i];
}
void NTT(int *a,int *b,int *c){
    int i;
    fo(i,0,len-1) A[i]=a[i],B[i]=b[i];
    DFT(A,1);DFT(B,1);
    fo(i,0,len-1) A[i]=(ll)A[i]*B[i]%mo;
    DFT(A,-1);
    fo(i,0,len-1) c[i]=A[i];
}
int main(){
    scanf("%d",&n);
    len=1;
    while (len<=n*2) len*=2;
    ni=quicksortmi(len,mo-2);
    ce=(db)log(len)/log(2);
    w[0]=1;
    w[1]=quicksortmi(GG,(mo-1)/len);
    fo(i,2,len) w[i]=(ll)w[i-1]*w[1]%mo;
    f[0]=1;
    fo(i,1,n) f[i]=(ll)f[i-1]*i%mo;
    g[n]=quicksortmi(f[n],mo-2);
    fd(i,n-1,0) g[i]=(ll)g[i+1]*(i+1)%mo;
    fo(i,0,n) a[i]=i%2?-g[i]:g[i];
    b[0]=1;
    b[1]=n+1;
    fo(i,2,n) b[i]=(ll)(quicksortmi(i,n+1)-1)*quicksortmi(i-1,mo-2)%mo*g[i]%mo;
    NTT(a,b,c);
    j=1;
    fo(i,0,n){
        if (i) j=(ll)j*i%mo;
        ans=(ans+(ll)quicksortmi(2,i)*f[i]%mo*c[i]%mo)%mo;
    }
    (ans+=mo)%=mo;
    printf("%d\n",ans);
}
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值