• LuoguP5748 集合划分计数


    题意

    一个有(n)个元素的集合,将其分为任意个非空子集,求方案数。集合之间是无序的,({{1,2},{3}}={{3},{1,2}})


    (f_n)表示用(n)个元素组成的集合的个数,显然(f_n=1)。设(F(x))(f)的指数型生成函数,那么(F(x)=sum_{i=1}frac{x^i}{i!})(F^i(x))的第(n)位就是(i)个元素个数之和为(n)的集合组合在一起的方案数。

    (g_i)(n=i)时的答案,再设(G(x))(g)的指数型生成函数。枚举划分的集合个数:

    [G(x)=sum_{i=0}frac{F^i(x)}{i!} ]

    显然(F(x)=e^x-1),那么(G(x)=e^{e^x-1})(G(x))(i)项乘(i!)就是(g_i)。多项式(exp)即可。

    #include<bits/stdc++.h>
    #define rg register
    #define il inline
    #define cn const
    #define gc getchar()
    #define fp(i,a,b) for(rg int i=(a),ed=(b);i<=ed;++i)
    #define fb(i,a,b) for(rg int i=(a),ed=(b);i>=ed;--i)
    using namespace std;
    typedef cn int cint;
    il int rd(){
       rg int x(0),f(1); rg char c(gc);
       while(c<'0'||'9'<c){ if(c=='-') f=-1; c=gc; }
       while('0'<=c&&c<='9') x=(x<<1)+(x<<3)+(c^48),c=gc;
       return x*f;
    }
    cint inf=0x3f3f3f3f,maxn=1000010,mod=998244353;
    
    int T,n=100000,inv[maxn],fac[maxn],ifac[maxn];
    int a[maxn],f[maxn],g[maxn],A[maxn],B[maxn];
    int lim=1,l,rev,r[maxn];
    
    il int fpow(int a,int b,int ans=1){
    	for(;b;b>>=1,a=1ll*a*a%mod)if(b&1)ans=1ll*ans*a%mod;
    	return ans;
    }
    il int finv(cint &n){return fpow(n,mod-2);}
    cint G=3,invG=finv(G);
    
    il void ntt(int *a,cint &f){
    	fp(i,0,lim-1)if(i<r[i])swap(a[i],a[r[i]]);
    	for(rg int md=1;md<lim;md<<=1){
    		rg int len=md<<1,Gn=fpow(f?G:invG,(mod-1)/len);
    		for(rg int l=0;l<lim;l+=len){
    			rg int Pow=1;
    			for(rg int nw=0;nw<md;++nw,Pow=1ll*Pow*Gn%mod){
    				rg int x=a[l+nw],y=1ll*Pow*a[l+nw+md]%mod;
    				a[l+nw]=(x+y)%mod,a[l+nw+md]=(x-y+mod)%mod;
    			}
    		}
    	}
    }
    
    void get_inv(int *a,int *b,int n){
    	if(n==1){b[0]=finv(a[0]);return;}
    	get_inv(a,b,n>>1);
    	fp(i,0,n)B[i]=a[i];
    	lim=1,l=0; while(lim<=n<<1)lim<<=1,++l; rev=finv(lim);
    	fp(i,0,lim-1)r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
    	ntt(B,1),ntt(b,1);
    	fp(i,0,lim)b[i]=1ll*b[i]*((2-1ll*B[i]*b[i]%mod+mod)%mod)%mod; ntt(b,0);
    	fp(i,0,n)b[i]=1ll*b[i]*rev%mod; fp(i,n+1,lim)b[i]=0;
    	fp(i,0,lim)B[i]=0;
    }
    
    il void get_ln(int *a,int *b,int n){
    	fp(i,1,n)A[i-1]=1ll*a[i]*i%mod; get_inv(a,b,n);
    	lim=1,l=0; while(lim<=n<<1)lim<<=1,++l; rev=finv(lim);
    	fp(i,0,lim-1)r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
    	ntt(A,1),ntt(b,1);
    	fp(i,0,lim)b[i]=1ll*b[i]*A[i]%mod; ntt(b,0);
    	fp(i,0,n)b[i]=1ll*b[i]*rev%mod; fp(i,n+1,lim)b[i]=0;
    	fp(i,0,lim)A[i]=0;
    	fb(i,n,1)b[i]=1ll*b[i-1]*inv[i]%mod; b[0]=0;
    }
    
    void get_exp(int *a,int *f,int n){
    	if(n==1){f[0]=1;return;}
    	get_exp(a,f,n>>1),get_ln(f,g,n);
    	g[0]=(1-g[0]+a[0]+mod)%mod; fp(i,1,n)g[i]=(a[i]-g[i]+mod)%mod;
    	lim=1,l=0; while(lim<=n<<1)lim<<=1,++l; rev=finv(lim);
    	fp(i,0,lim-1)r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
    	ntt(f,1),ntt(g,1);
    	fp(i,0,lim)f[i]=1ll*f[i]*g[i]%mod; ntt(f,0);
    	fp(i,0,n)f[i]=1ll*f[i]*rev%mod; fp(i,n+1,lim)f[i]=0;
    	fp(i,0,lim)g[i]=0;
    }
    
    int main(){
    	fac[0]=1; fp(i,1,n)fac[i]=1ll*fac[i-1]*i%mod;
    	inv[1]=1; fp(i,2,n)inv[i]=1ll*(mod-mod/i)*inv[mod%i]%mod;
    	ifac[0]=1; fp(i,1,n)ifac[i]=1ll*ifac[i-1]*inv[i]%mod;
    	
    	fp(i,1,n)a[i]=ifac[i];
    	while(lim<=n)lim<<=1; get_exp(a,f,lim);
    	
    	T=rd();
    	while(T--)n=rd(),printf("%lld
    ",1ll*f[n]*fac[n]%mod);
    	return 0;
    }
    
  • 相关阅读:
    B-线性代数-距离公式汇总
    B-线性代数-范数
    B-线性代数-矩阵转置
    B-概率论-贝叶斯决策
    B-概率论-极大似然估计
    B-概率论-条件概率
    2018.1.7java转型
    追求
    面向心态
    数据类型和type函数
  • 原文地址:https://www.cnblogs.com/akura/p/12256470.html
Copyright © 2020-2023  润新知