首页 > 技术文章 > 【bzoj4596】[Shoi2016]黑暗前的幻想乡 (矩阵树定理+容斥)

ezoiLZH 2018-07-30 11:31 原文

Description

四年一度的幻想乡大选开始了,最近幻想乡最大的问题是很多来历不明的妖怪涌入了幻想乡,扰乱了幻想乡昔日的秩序。但是幻想乡的建制派妖怪(人类)博丽灵梦和八云紫等人整日高谈所有妖怪平等,幻想乡多元化等等,对于幻想乡目前面临的种种大问题却给不出合适的解决方案。
风间幽香是幻想乡里少有的意识到了问题的严重性的大妖怪。她这次勇敢的站了出来参加幻想乡大选。提出包括在幻想乡边境建墙(并让人类出钱),大力开展基础设施建设挽回失业率等一系列方案,成为了大选年出人意料的黑马并顺利的当上了幻想乡的大统领。
幽香上台以后,第一项措施就是要修建幻想乡的公路。幻想乡有 N 个城市,之间原来没有任何路。幽香向选民承诺要减税,所以她打算只修 N- 1 条路将这些城市连接起来。但是幻想乡有正好 N- 1 个建筑公司,每个建筑公司都想在修路的过程中获得一些好处。
虽然这些建筑公司在选举前没有给幽香钱,幽香还是打算和他们搞好关系,因为她还指望他们帮她建墙。所以她打算让每个建筑公司都负责一条路来修。每个建筑公司都告诉了幽香自己有能力负责修建的路是哪些城市之间的。所以幽香打算选择 N-1 条能够连接幻想乡所有城市的边,然后每条边都交给一个能够负责该边的建筑公司修建,并且每个建筑公司都恰好修一条边。幽香现在想要知道一共有多少种可能的方案呢?两个方案不同当且仅当它们要么修的边的集合不同,要么边的分配方式不同。

Input

第一行包含一个正整数 N(N<=17), 表示城市个数。
接下来 N-1 行,其中第 i行表示第 i个建筑公司可以修建的路的列表:
以一个非负数mi 开头,表示其可以修建 mi 条路,接下来有mi 对数,
每对数表示一条边的两个端点。其中不会出现重复的边,也不会出现自环。

Output

仅一行一个整数,表示所有可能的方案数对 10^9 + 7 取模的结果。

 题解:

这题叫我们给出 n 个点和 n−1 种颜色,每种颜色有若干条边。求这张图多少棵每种颜色的边都出现过的生成树

这首先可以看出,矩阵树定理,但要求每种颜色都取,我们考虑容斥。

$$ ans= f_n-f_{n-1}+f_{n-2}-f_{n-3}+ \ ... $$

f表示选其中 x 个颜色。

然后我们可以使用状压,1 表示选这个颜色,0 表示不选这个颜色。

CODE:

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<vector>
 4 #include<cstring>
 5 using namespace std;
 6 
 7 #define mod 1000000007
 8 int n,m,x,y,ans=0,a[20][20],siz[20];
 9 struct Edge{
10     int x,y;
11     Edge(int a,int b){x=a,y=b;}
12     Edge(){x=y=0;}
13 }e[20][500];
14 
15 int qpow(int x,int y){
16     int ans=1;
17     while(y){
18         if(y&1)ans=1LL*ans*x%mod;
19         y>>=1,x=1LL*x*x%mod;
20     }
21     return ans;
22 }
23 
24 int Gauss(){
25     for(int i=1;i<n;i++){
26         for(int j=i+1;j<n;j++){
27             if(a[i][i]==0)return 0;
28             if(a[j][i]==0)continue;
29             int t=1LL*a[j][i]*qpow(a[i][i],mod-2)%mod;
30             for(int k=i;k<n;k++)
31             a[j][k]=(a[j][k]-1LL*a[i][k]*t%mod+mod)%mod;
32         }
33     }
34     int ans=1;
35     for(int i=1;i<n;i++)ans=1LL*ans*a[i][i]%mod;
36     return ans;
37 }
38 
39 int main(){
40     scanf("%d",&n);
41     for(int i=1;i<n;i++){
42         scanf("%d",siz+i);
43         for(int j=1;j<=siz[i];j++){
44             scanf("%d%d",&x,&y);
45             e[i][j]=Edge(x,y);
46         }
47     }
48     for(int i=0;i<(1<<n-1);i++){
49         memset(a,0,sizeof(a));
50         int f=1;
51         for(int j=1;j<n;j++){
52             if(i&(1<<j-1)){
53                 for(int k=1;k<=siz[j];k++){
54                     a[e[j][k].x][e[j][k].y]--;
55                     a[e[j][k].y][e[j][k].x]--;
56                     a[e[j][k].x][e[j][k].x]++;
57                     a[e[j][k].y][e[j][k].y]++;
58                 }
59             }else f=-f;
60         }
61         for(int j=1;j<=n;j++)
62         for(int k=1;k<=n;k++)if(a[j][k]<0)a[j][k]+=mod;
63         ans+=f*Gauss();
64         if(ans<0)ans+=mod;
65         if(ans>=mod)ans-=mod;
66     }
67     printf("%d",ans);
68 }

推荐阅读