题解:【bzoj4596】[Shoi2016]黑暗前的幻想乡(状压+容斥原理+矩阵树定理+高斯消元)

Description
四年一度的幻想乡大选开始了,最近幻想乡最大的问题是很多来历不明的妖

怪涌入了幻想乡,扰乱了幻想乡昔日的秩序。但是幻想乡的建制派妖怪(人类)

博丽灵梦和八云紫等人整日高谈所有妖怪平等,幻想乡多元化等等,对于幻想乡

目前面临的种种大问题却给不出合适的解决方案。

风间幽香是幻想乡里少有的意识到了问题的严重性的大妖怪。她这次勇敢的

站了出来参加幻想乡大选。提出包括在幻想乡边境建墙(并让人类出钱),大力

开展基础设施建设挽回失业率等一系列方案,成为了大选年出人意料的黑马并顺

利的当上了幻想乡的大统领。

幽香上台以后,第一项措施就是要修建幻想乡的公路。幻想乡有 N 个城市,

之间原来没有任何路。幽香向选民承诺要减税,所以她打算只修 N- 1 条路将

这些城市连接起来。但是幻想乡有正好 N- 1 个建筑公司,每个建筑公司都想

在修路的过程中获得一些好处。

虽然这些建筑公司在选举前没有给幽香钱,幽香还是打算和他们搞好关系,

因为她还指望他们帮她建墙。所以她打算让每个建筑公司都负责一条路来修。

每个建筑公司都告诉了幽香自己有能力负责修建的路是哪些城市之间的。所

以幽香打算选择 N-1 条能够连接幻想乡所有城市的边,然后每条边都交给一

个能够负责该边的建筑公司修建,并且每个建筑公司都恰好修一条边。

幽香现在想要知道一共有多少种可能的方案呢?两个方案不同当且仅当它

们要么修的边的集合不同,要么边的分配方式不同。

Input
第一行包含一个正整数 N(N<=17), 表示城市个数。

接下来 N-1 行,其中第 i行表示第 i个建筑公司可以修建的路的列表:

以一个非负数mi 开头,表示其可以修建 mi 条路,接下来有mi 对数,

每对数表示一条边的两个端点。其中不会出现重复的边,也不会出现自环。

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

Sample Input
4
2 3 2 4 2
5 2 1 3 1 3 2 4 1 4 3
4 2 1 3 2 4 1 4 2

Sample Output
17

这道题的解法十分暴力。

我们发现 n 的范围很小,所以我们可以考虑状压。

每一个状态 i 表示当前计算的生成树数量包含哪些颜色

根据容斥原理可得:最后的答案就是:任取 一种颜色不选 + 两种颜色不选 ……

统计每个状态的答案用矩阵树定理+高斯消元即可

时间复杂度 O ( 2 n n 3 )
代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int mod=1e9+7;
int n,ans,m[20],edge[410][410][2],a[20][20];
int qpow(int x,int n){
    int ret=1;
    while(n){
        if(n&1)
            ret=1ll*ret*x%mod;
        x=1ll*x*x%mod;
        n>>=1;
    }
    return ret;
}
int gauss(int n){
    int ret=1;
    for(int i=1;i<n;i++){
        for(int j=i+1;j<=n;j++){
            if(!a[i][i])
                return 0;
            if(!a[j][i])
                continue;
            int tmp=1ll*a[j][i]*qpow(a[i][i],mod-2)%mod;
            for(int k=i;k<=n;k++){
                a[j][k]-=1ll*a[i][k]*tmp%mod;
                if(a[j][k]<0)
                    a[j][k]+=mod;
            }
        }
    }
    for(int i=1;i<=n;i++)
        ret=1ll*ret*a[i][i]%mod;
    return ret;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        scanf("%d",m+i);
        for(int j=1;j<=m[i];j++)
            scanf("%d%d",edge[i][j],edge[i][j]+1);
    }
    for(int i=0;i<(1<<(n-1));i++){
        memset(a,0,sizeof(a));
        int f=1;
        for(int j=1;j<n;j++){
            if(i&(1<<(j-1))){
                for(int k=1;k<=m[j];k++){
                    a[edge[j][k][0]][edge[j][k][1]]--;
                    a[edge[j][k][1]][edge[j][k][0]]--;
                    a[edge[j][k][0]][edge[j][k][0]]++;
                    a[edge[j][k][1]][edge[j][k][1]]++;
                }
            }
            else f=-f;
        }
        for(int j=1;j<=n;j++)
            for(int k=1;k<=n;k++)
                if(a[j][k]<0)
                    a[j][k]+=mod;
        ans+=f*gauss(n-1);
        if(ans<0)
            ans+=mod;
        if(ans>=mod)
            ans-=mod;
    }
    printf("%d",ans);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/ezoiHQM/article/details/81258462