bzoj5153 [Wc2018]州区划分

题目链接

 

正解:子集和变换。

考场上只会暴力和$p=0$的情况,还只会$O(2^{n}*n^{3})$的。

然而这题题面出锅,导致考场上一直在卡裸暴力,后面的部分分没写了。。听$laofu$说$O(2^{n}*n^{3})$可以过。。

所以直接讲正解。。

我们假设每个城市可以在两个不同集合,那么可以把子集卷积变成或卷积。

我们只要记下当前总共有多少个点,于是考虑设$f[i][S]$表示$i$个点,集合为$S$的方案数。

最后的$f[n][all]$就是答案,显然这个状态中的每个城市只会出现一次。

那么$f[i][S]=\sum f[j][T]*(\frac{sum[A]}{sum[S]})^{p}$,其中$A|T=S$,且$A$是一个合法集合。

可以把分母移项到左边,然后我们可以设$g[i][S]$表示如果$S$是一个合法集合,且$S$的位数为$i$,那么$g[i][S]=sum[S]^{p}$,否则为$0$。

那么$f[i][S]*sum[S]^{p}=\sum f[j][T]*g[i-j][A]$。注意到这个式子可以直接$FMT$以后点乘,再$IFMT$回来以后作除法得到,总复杂度为$O(2^{n}*n^{2})$。

感觉这道题其实并没有那么难,但是考场上被题面以及固定的套路给局限住了,所以并没有想到可以交换$dp$的两维状态从而优化复杂度。

 

 1 #include <bits/stdc++.h>
 2 #define il inline
 3 #define RG register
 4 #define ll long long
 5 #define rhl (998244353)
 6 #define N (1<<21|1)
 7 
 8 using namespace std;
 9 
10 int fa[25],g[25],w[N],cnt[N],can[N],inv[10005],n,m,p,all;
11 int f[22][N],h[22][N];
12 
13 il int gi(){
14   RG int x=0,q=1; RG char ch=getchar();
15   while ((ch<'0' || ch>'9') && ch!='-') ch=getchar();
16   if (ch=='-') q=-1,ch=getchar();
17   while (ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
18   return q*x;
19 }
20 
21 il int qpow(RG int a,RG int b){
22   if (!b) return 1;
23   if (b==1) return a;
24   return 1LL*a*a%rhl;
25 }
26 
27 il void fmt(int *a){
28   for (RG int i=1;i<all;i<<=1)
29     for (RG int j=0;j<all;++j){
30       if (j&i) a[j]+=a[j^i];
31       if (a[j]>=rhl) a[j]-=rhl;
32     }
33   return;
34 }
35 
36 il void ifmt(int *a){
37   for (RG int i=1;i<all;i<<=1)
38     for (RG int j=0;j<all;++j){
39       if (j&i) a[j]-=a[j^i];
40       if (a[j]<0) a[j]+=rhl;
41     }
42   return;
43 }
44 
45 il int find(RG int x){
46   return fa[x]==x ? x : fa[x]=find(fa[x]);
47 }
48 
49 int main(){
50 #ifndef ONLINE_JUDGE
51   freopen("walk.in","r",stdin);
52   freopen("walk.out","w",stdout);
53 #endif
54   n=gi(),m=gi(),p=gi(),all=1<<n,inv[1]=1;
55   for (RG int i=1,u,v;i<=m;++i)
56     u=gi()-1,v=gi()-1,g[u]|=1<<v,g[v]|=1<<u;
57   for (RG int i=2;i<=3000;++i)
58     inv[i]=1LL*(rhl-rhl/i)*inv[rhl%i]%rhl;
59   for (RG int i=0;i<n;++i) w[1<<i]=gi(); fmt(w);
60   for (RG int i=1;i<all;++i) cnt[i]=cnt[i>>1]+(i&1);
61   for (RG int i=1,fg,lst;i<all;++i){
62     fg=0,lst=-1;
63     for (RG int j=0;j<n;++j) fa[j]=j;
64     for (RG int j=0;j<n;++j){
65       if (!(i>>j&1)) continue;
66       for (RG int k=0,x,y;k<n;++k){
67     if (!(i>>k&1) || !(g[j]>>k&1)) continue;
68     x=find(j),y=find(k); if (x!=y) fa[x]=y;
69       }
70     }
71     for (RG int j=0;j<n;++j){
72       if (i>>j&1){
73     if (lst==-1) lst=find(j);
74     else if (lst!=find(j)){ fg=1; break; }
75       }
76       if ((i>>j&1) && (cnt[g[j]&i]&1)){ fg=1; break; }
77     }
78     can[i]=fg;
79   }
80   for (RG int i=0;i<all;++i) if (can[i]) h[cnt[i]][i]=qpow(w[i],p);
81   for (RG int i=1;i<=n;++i) fmt(h[i]); f[0][0]=1,fmt(f[0]);
82   for (RG int i=1;i<=n;++i){
83     int *F=f[i];
84     for (RG int j=0;j<i;++j){
85       int *a=f[j],*b=h[i-j];
86       for (RG int s=0;s<all;++s)
87     F[s]=(1LL*a[s]*b[s]+F[s])%rhl;
88     }
89     ifmt(F);
90     for (RG int s=0;s<all;++s)
91       F[s]=i==cnt[s]?1LL*F[s]*qpow(inv[w[s]],p)%rhl:0;
92     if (i^n) fmt(F);
93   }
94   cout<<f[n][all-1]; return 0;
95 }

 

posted @ 2018-02-11 19:51  wfj_2048  阅读(607)  评论(0编辑  收藏  举报