BZOJ4596 || 洛谷P4336 [SHOI2016]黑暗前的幻想乡【容斥+矩阵树定理】

Time Limit: 20 Sec
Memory Limit: 256 MB

Description

四年一度的幻想乡大选开始了，最近幻想乡最大的问题是很多来历不明的妖怪涌入了幻想乡，扰乱了幻想乡昔日的秩序。但是幻想乡的建制派妖怪（人类）博丽灵梦和八云紫等人整日高谈所有妖怪平等，幻想乡多元化等等，对于幻想乡目前面临的种种大问题却给不出合适的解决方案。

风间幽香是幻想乡里少有的意识到了问题的严重性的大妖怪。她这次勇敢的站了出来参加幻想乡大选。提出包括在幻想乡边境建墙（并让人类出钱），大力开展基础设施建设挽回失业率等一系列方案，成为了大选年出人意料的黑马并顺利的当上了幻想乡的大统领。

幽香上台以后，第一项措施就是要修建幻想乡的公路。幻想乡有 N 个城市，之间原来没有任何路。幽香向选民承诺要减税，所以她打算只修 N- 1 条路将这些城市连接起来。但是幻想乡有正好 N- 1 个建筑公司，每个建筑公司都想在修路的过程中获得一些好处。

虽然这些建筑公司在选举前没有给幽香钱，幽香还是打算和他们搞好关系，因为她还指望他们帮她建墙。所以她打算让每个建筑公司都负责一条路来修。每个建筑公司都告诉了幽香自己有能力负责修建的路是哪些城市之间的。所以幽香打算选择 N-1 条能够连接幻想乡所有城市的边，然后每条边都交给一个能够负责该边的建筑公司修建，并且每个建筑公司都恰好修一条边。

幽香现在想要知道一共有多少种可能的方案呢？两个方案不同当且仅当它们要么修的边的集合不同，要么边的分配方式不同。

Input

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

Output

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

---

题目分析

假如直接用题目给出的所有边构造矩阵套用矩阵树定理
那么其中可能包含了一些不是所有施工队都参加了的生成树方案

所以考虑把这些没有某些施工队参加的方案数减去，即不考虑某一个施工队来构造矩阵
但是又发现这其中有些方案重合了，所以加回来…就这样反复容斥

老套路二进制枚举容斥，复杂度$O(n^3{2}^n)$

#include<iostream>
#include<cstdio>
#include<cmath>
#include<queue>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long lt;

int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return f*x;
}

const lt mod=1e9+7;
const int maxn=20;
int n;
lt a[maxn][maxn];
int cnt[maxn];
struct node{int u,v;}rem[maxn][maxn*100];

int gauss()
{
    lt res=1;
    for(int i=1;i<n;++i)
    {
        for(int j=i+1;j<n;++j)
        while(a[j][i])
        {
            lt t=a[i][i]/a[j][i];
            for(int k=i;k<n;++k)
            a[i][k]=(a[i][k]-t*a[j][k]+mod)%mod;
            
            swap(a[j],a[i]);
            res=-res;
        }
        res=(res*a[i][i])%mod;
    }
    return (res+mod)%mod;
}

int main() 
{ 	
    n=read();
    for(int i=1;i<n;++i)
    {
        cnt[i]=read();
        for(int j=1;j<=cnt[i];++j)
        {
            int u=read(),v=read();
            rem[i][j]=(node){u,v};
        }
    }
    
    lt ans=0;
    int lim=(1<<(n-1))-1;
    for(int i=0;i<=lim;++i)
    {
        memset(a,0,sizeof(a));
        
        int k=1;
        for(int j=0;j<n-1;++j)
        {
            if(!(i&(1<<j)))
            {
                for(int l=1;l<=cnt[j+1];++l)
                {
                    int u=rem[j+1][l].u,v=rem[j+1][l].v;
                    a[u][u]++; a[v][v]++; a[u][v]--; a[v][u]--;
                }
            }
            else k=-k;
        }
        
        ans=(ans+k*gauss())%mod;
    }
    
    printf("%lld",(ans+mod)%mod);
    return 0;
}