[LOJ#2340] [WC2018] 州区划分

题目链接

洛谷题面。

LOJ题面。还是LOJ机子比较快

Solution

设(f(s))表示选(s)这些城市的总代价，那么我们可以得到一个比较显然的(dp)：

[f(s)=frac{1}{sum_s^p}sum_{tsubset s} f(t)g(s-t) ]

其中(g(i))表示合法的(sum_i^p)，即若不合法则为(0)。

这个(dp)是(O(3^n))的，注意到这是个子集卷积的形式，我们可以考虑用(fwt)优化它。

那么我们可以一层一层的(dp)，即按照(bitcnt(s))的顺序从小到大计算(f)，假设当前算到了(bitcnt=i)，那么那些(bitcnt<i)的(f)肯定已经算完了，所以我们可以用(fwt)加速来算好当前层的(f)。

容易发现这样做的复杂度是(O(n^22^n))的，足以通过此题。

洛谷机子好慢啊...要开O2才能过，loj一半时限就过了

#include<bits/stdc++.h>
using namespace std;
 
void read(int &x) {
    x=0;int f=1;char ch=getchar();
    for(;!isdigit(ch);ch=getchar()) if(ch=='-') f=-f;
    for(;isdigit(ch);ch=getchar()) x=x*10+ch-'0';x*=f;
}
 
void print(int x) {
    if(x<0) putchar('-'),x=-x;
    if(!x) return ;print(x/10),putchar(x%10+48);
}
void write(int x) {if(!x) putchar('0');else print(x);putchar('
');}

#define lf double
#define ll long long 

const int maxn = (1<<21)+10;
const int inf = 1e9;
const lf eps = 1e-8;
const int mod = 998244353;

int cnt[maxn],sum[maxn],g[22][maxn];
int f[22][maxn],n,m,p,w[30],fa[30],d[30],head[30],tot,inv[maxn];
struct edge{int to,nxt;}e[900];

void Add(int u,int v) {e[++tot]=(edge){v,head[u]},head[u]=tot;}
void ins(int u,int v) {Add(u,v),Add(v,u);}

int find(int x) {return fa[x]==x?x:fa[x]=find(fa[x]);}

int add(int x,int y) {return x+y>mod?x+y-mod:x+y;}
int del(int x,int y) {return x-y<0?x-y+mod:x-y;}
int mul(int x,int y) {return 1ll*x*y-1ll*x*y/mod*mod;}

int check(int s) {
	if(cnt[s]==1) return 0;
	for(int i=1;i<=n;i++) fa[i]=i,d[i]=0;
	for(int x=1;x<=n;x++)
		if(s>>(x-1)&1) {
			for(int u,v,i=head[x];i;i=e[i].nxt) {
				if(!(s>>(e[i].to-1)&1)) continue;
			    d[e[i].to]++;
				if((u=find(x))!=(v=find(e[i].to))) fa[u]=v; 
			}
		}
	int rt=__builtin_ctz(s&-s)+1;
	for(int i=1;i<=n;i++) if(s>>(i-1)&1) if(find(i)!=find(rt)||(d[i]&1)) return 1;
	return 0;
}

void fwt(int *r,int N) {
	for(int i=1;i<N;i<<=1)
		for(int j=0;j<N;j+=i<<1)
			for(int k=0;k<i;k++)
				r[i+j+k]=add(r[i+j+k],r[j+k]);
}

void ifwt(int *r,int N) {
	for(int i=1;i<N;i<<=1)
		for(int j=0;j<N;j+=i<<1)
			for(int k=0;k<i;k++)
				r[i+j+k]=del(r[i+j+k],r[j+k]);
}

int qpow(int a,int x) {
	int res=1;
	for(;x;x>>=1,a=mul(a,a)) if(x&1) res=mul(res,a);
	return res;
}

int main() {
	read(n),read(m),read(p);
	for(int i=1,x,y;i<=m;i++) read(x),read(y),ins(x,y);
	for(int i=1;i<=n;i++) read(w[i]);
	int all=1<<n;
	for(int i=1;i<all;i++) cnt[i]=__builtin_popcount(i);
	for(int i=1;i<all;i++) sum[i]=w[__builtin_ctz(i&-i)+1]+sum[i^(i&-i)];
	for(int i=1;i<all;i++) sum[i]=qpow(sum[i],p);
	for(int i=0;i<all;i++) g[cnt[i]][i]=sum[i]*check(i),inv[i]=qpow(sum[i],mod-2);
	for(int i=0;i<=n;i++) fwt(g[i],all);
	f[0][0]=1;fwt(f[0],all);
	for(int i=1;i<=n;i++) {
		for(int j=0;j<i;j++)
			for(int s=0;s<all;s++)
				f[i][s]=add(f[i][s],mul(f[j][s],g[i-j][s]));
		ifwt(f[i],all);
		for(int s=0;s<all;s++) f[i][s]=cnt[s]==i?mul(f[i][s],inv[s]):0;
		if(i!=n) fwt(f[i],all);
	}
	write(f[n][all-1]);
	return 0;
}