luogu P4221 [WC2018]州区划分

题面传送门
判欧拉回路写错调了一个多小时
首先可以\(O(2^nm)\)处理出哪些是可以选的就有了\(O(n3^n)\)的爆枚子集的做法。
不难想到子集卷积,但是子集卷积是卷两个没有关系的数组,但是这个是自己卷自己。
容易发现一个数枚举子集的过程是不可能从二进制和它个数相同的数拿答案的,这就意味着同一层算答案的时候是互不干扰的。
所以可以直接卷。
然后还有一个问题就是要除一个数。
很抱歉我并不会什么奇怪的科技,所以只能写那种FWT回去除了再FWT回来的东西。
时间复杂度\(O(2^nn^2)\)
code:

#include<bits/stdc++.h>
#define I inline
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define abs(x) ((x)>0?(x):-(x))
#define re register
#define RI re int
#define ll long long
#define db double
#define lb long db
#define N 21
#define K (1<<N)
#define mod 998244353
#define Mod 998244352
#define eps (1e-4)
#define U unsigned int
#define it iterator
#define Gc() getchar() 
#define Me(x,y) memset(x,y,sizeof(x))
#define d(x,y) (n*(x-1)+(y))
using namespace std;
int n,m,k,p,x[N*N+5],y[N*N+5],In[N+5],W[K+5],H[K+5],cnt,un,wn,Fl,lg[K+5],fa[K+5],F1[N+1][K+5],F2[N+1][K+5];ll G[K+5],Inv[K+5];
I ll mpow(ll x,int y=mod-2){ll Ans=1;while(y) y&1&&(Ans=Ans*x%mod),y>>=1,x=x*x%mod;return Ans;}
I void OR(int *A,int n,int Fl){
	RI i,j,h;for(i=2;i<=n;i<<=1){
		for(j=0;j<n;j+=i) for(h=j;h<j+i/2;h++) A[h+i/2]=(1ll*A[h+i/2]+mod+A[h]*Fl)%mod;
	}
}I int Getfa(int x){return x==fa[x]?x:fa[x]=Getfa(fa[x]);}
I void merge(int x,int y){un=Getfa(x);wn=Getfa(y);un^wn&&(fa[un]=wn,cnt--);}
int main(){
	freopen("1.in","r",stdin);
	RI i,j,h;scanf("%d%d%d",&n,&m,&p);k=1<<n;for(i=1;i<=m;i++) scanf("%d%d",&x[i],&y[i]);for(i=1;i<=n;i++)scanf("%d",&W[i]);for(i=1;i<k;i++) lg[i]=lg[i/2]+1;
	for(i=1;i<k;i++)H[i]=H[i>>1]+(i&1),G[i]=(G[i^(i&-i)]+W[lg[i&-i]])%mod;
	for(i=0;i<k;i++){
		Me(In,0);for(j=1;j<=n;j++) fa[j]=j;cnt=H[i];G[i]=mpow(G[i],p);Inv[i]=G[i]?mpow(G[i]):1;for(j=1;j<=m;j++) i>>(x[j]-1)&1&&i>>(y[j]-1)&1&&(merge(x[j],y[j]),In[x[j]]^=1,In[y[j]]^=1);
		for(Fl=j=1;j<=n;j++) i>>(j-1)&1&&(Fl&=In[j]^1);F1[H[i]][i]=G[i]*(Fl^1||cnt^1);
	}F2[0][0]=1;OR(F2[0],k,1);for(i=1;i<=n;i++){
		OR(F1[i],k,1);for(j=1;j<=i;j++) for(h=0;h<k;h++) F2[i][h]=(F2[i][h]+1ll*F1[j][h]*F2[i-j][h])%mod;OR(F2[i],k,-1);
		for(j=0;j<k;j++) F2[i][j]=1ll*F2[i][j]*Inv[j]%mod;OR(F2[i],k,1);
	}OR(F2[n],k,-1);printf("%d\n",F2[n][k-1]);
}
上一篇:大数据-数据分析-numpy库-数组形状操作-reshape、resize、fl、数组的叠加


下一篇:修复DNA