「NOI十联测」黑暗

「NOI十联测」黑暗

\(n\) 个点的无向图,每条边都可能存在,一个图的权值是连通块个数的 \(m\) 次方,求所有可能的图的权值和。(n≤30000,m≤15)

\(ans[n][m]\)为n个点,图的权值是连通块个数的 m 次方,最终的答案.

\(g[i]\)\(i\)个点无向联通图数目.

一个图,若连通块数T增加1,那么贡献为\(\displaystyle (T+1)^m=\sum_{i=0}^m C_m^i T^i\).

这里选取\(i\)个点组成一个连通块(包含一个指定的点),联通块数目增加1.

\[ans[n][m]=\sum_{i=1}^{n} g_i C_{n-1}^{i-1} (\sum_{j=0}^m C_m^j ans[n-i][j]) \]

\[ans[n][m]=m!(n-1)! \sum_{i=1}^{n} {g_i \over (i-1)!}{{\sum_{j=0}^m {ans[n-i][j]\over j! (m-j)!}} \over (n-i)!} \]

\(\displaystyle T[i][j]=\sum_{k=0}^j {ans[i][k]\over k! (j-k)!}\)

\[ans[n][m]=m!(n-1)! \sum_{i=1}^{n} {g_i \over (i-1)!}{ T[n-i][m] \over (n-i)!} \]

枚举m.

m=0的时候ans即为图的个数.

之后分治FFT...

复杂度\(O(nmlog(n^2)+nm^2)\).

代码里把\(i=n\)单独提出,其实可以不提出。

#include<bits/stdc++.h>
#define rep(q,a,b) for(int q=a,q##_end_=b;q<=q##_end_;++q)
#define dep(q,a,b) for(int q=a,q##_end_=b;q>=q##_end_;--q)
#define mem(a,b) memset(a,b,sizeof a )
#define debug(a) cerr<<#a<<' '<<a<<"___"<<endl
using namespace std;
void in(int &r) {
	static char c;
	r=0;
	while(c=getchar(),!isdigit(c));
	do r=(r<<1)+(r<<3)+(c^48);
	while(c=getchar(),isdigit(c));
}
const int mn=30005;
const int mod=998244353;
int mul[mn],inv[mn];
int all[mn],un_able[mn],_able[mn];
int mlv(int x,int v){
    int ans=1;
    while(v){
	    if(v&1)ans=1LL*ans*x%mod;
	    x=1LL*x*x%mod,v>>=1;
	}
	return ans;
}
namespace NTT{
    const int g=3;
    const int gg=332748118; 
    int n,lim,to[66000];
    void DFT(int* a,int iv){
	    rep(q,0,n-1)if(to[q]<q)swap(a[q],a[to[q]]);
	    for(int len=2;len<=n;len<<=1){
		    int mv=1,ml=mlv(iv?g:gg,(mod-1)/len),sp=len>>1;
		    for(int k=0;k<sp;++k){
			    for(int* p=a;p!=a+n;p+=len){
				    int mid=1LL*p[k+sp]*mv%mod;
				    p[k+sp]=(p[k]-mid)%mod;
				    p[k]=(p[k]+mid)%mod;
				}
				mv=1LL*mv*ml%mod;
			}
		}
	}
	void solve(int* a,int len1,int* b,int len2,int* c){
	    n=1,lim=0;
	    while(n<(len1+len2))n<<=1,++lim;
	    rep(q,len1,n-1)a[q]=0;
	    rep(q,len2,n-1)b[q]=0;
	    rep(q,0,n-1)to[q]=to[q>>1]>>1 | (q&1)<<(lim-1);
	    DFT(a,1),DFT(b,1);
	    rep(q,0,n-1)c[q]=1LL*a[q]*b[q]%mod;
	    DFT(c,0);
	    int inv_n=mlv(n,mod-2);
	    rep(q,0,n-1)c[q]=1LL*c[q]*inv_n%mod;
	    rep(q,0,n-1)c[q]=(c[q]+mod)%mod;
	}
}
int dp[mn],dp1[mn],A[66000],B[66000],C[66000];
void solve(int l,int r){
	if(l>r)return;
    if(l==r){
    	un_able[l]=1LL*un_able[l]*mul[l-1]%mod;
	    _able[l]=((long long)all[l]-un_able[l]+mod)%mod;
	    return;
	}
	int mid=l+r>>1;
	solve(l,mid);
	rep(q,l,mid)A[q-l]=1LL*_able[q]*inv[q-1]%mod;
	rep(q,1,r-l)B[q-1]=1LL*all[q]*inv[q]%mod;
	NTT::solve(A,mid-l+1,B,r-l,C);
	rep(q,mid+1,r)un_able[q]=(un_able[q]+C[q-l-1])%mod;
	solve(mid+1,r);
}
int ans[16][30005],mid[16][30005],nw;
int *md,*md1;
void cdq(int l,int r){
    if(l>r)return;
    if(l==r){
	    md[l]=1LL*mul[nw]*mul[l-1]%mod*md[l]%mod;
	    md[l]=(md[l]+_able[l])%mod;
	    rep(q,0,nw)md1[l]=(md1[l]+1LL*ans[q][l]*inv[q]%mod*inv[nw-q])%mod;
	    return;
	}
	int mid=l+r>>1;
	cdq(l,mid);
	rep(q,l,mid)A[q-l]=1LL*md1[q]*inv[q]%mod;
	rep(q,1,r-l)B[q-1]=1LL*_able[q]*inv[q-1]%mod;
	NTT::solve(A,mid-l+1,B,r-l,C);
	rep(q,mid+1,r)md[q]=(md[q]+C[q-l-1])%mod;
	cdq(mid+1,r);
}
struct qr{
    int a,b;
}st[1005];
int main(){
	freopen("dark.in","r",stdin);
	freopen("dark.out","w",stdout);
    
	
	int T,m,n=15;
	in(T);
	rep(q,1,T)in(st[q].a),in(st[q].b),n=max(n,st[q].a);
	
	mul[0]=1;
	rep(q,1,n)mul[q]=1LL*mul[q-1]*q%mod;
	inv[0]=inv[1]=1;
	rep(q,2,n)inv[q]=1LL*(mod-mod/q)*inv[mod%q]%mod;
	rep(q,1,n)inv[q]=1LL*inv[q]*inv[q-1]%mod;
	
	all[0]=1;
	rep(q,1,n)all[q]=mlv(2,q*(q-1)>>1);
	
	solve(1,n);
	
	md=ans[0],md1=mid[0];
	rep(q,1,n)md[q]=md1[q]=mlv(2,q*(q-1)>>1);
	rep(q,1,15){
		md=ans[q],md1=mid[q],nw=q;
	    cdq(1,n);
	}
	
	rep(q,1,T)printf("%d\n",ans[st[q].b][st[q].a]);
	return 0;
}
posted @ 2019-08-01 16:37  Eeis  阅读(129)  评论(0编辑  收藏  举报