[BZOJ4596][SHOI2016]黑暗前的幻想乡（矩阵树定理+容斥）

题目：

我是超链接

题意：

n个点要修n-1条路（形成一棵树）。有n-1个公司，每个公司可以修建某些路径。求每个公司恰好修建一条路能修成树的方案数。

题解：

生成树的数量要考虑矩阵树定理，但是如果把所有的边加进去，不能保证每个公司都修一条路。
我们考虑容斥，其实答案就是：至少不选0个公司-至少不选1个公司+至少不选2个公司……
我们可以二进制枚举哪些公司选，剩下的就是不选的。

代码：

#include <cstdio>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
#define LL long long 
using namespace std;
const int mod=1e9+7;
struct hh{int x,y;hh(int X=0,int Y=0) {x=X;y=Y;}};
vector <hh> e[20];LL a[25][25];int m[25];
LL ksm(LL a,LL k)
{
    LL ans=1;
    for (;k;k>>=1,a=a*a%mod)
      if (k&1) ans=ans*a%mod;
    return ans;
}
LL gauss(int n)
{
    LL ans=1;int flag=0;
    for (int i=1;i<=n;i++)
      for (int j=1;j<=n;j++) a[i][j]%=mod;
    for (int i=1;i<=n;i++)
    {
        int num=i;
        for (int j=i;j<=n;j++)
          if (abs(a[num][i])<abs(a[j][i])) num=j;
        if (!a[num][i]) return 0;
        if (num!=i)
        {
            for (int j=i;j<=n;j++) swap(a[num][j],a[i][j]);
            flag^=1;
        }
        for (int j=i+1;j<=n;j++)
          while (a[j][i])
          {
            LL t=a[j][i]*ksm(a[i][i],mod-2)%mod;
            for (int k=i;k<=n;k++) a[j][k]=(a[j][k]-t*a[i][k]%mod+mod)%mod;
            if (!a[j][i]) break;flag^=1;
            for (int k=i;k<=n;k++) swap(a[i][k],a[j][k]);
          }
        ans=ans*a[i][i]%mod;
    }
    if (flag) ans=-ans;
    return (ans+mod)%mod;
}
int main()
{
    int n;scanf("%d",&n);
    for (int i=1;i<n;i++)
    {
        scanf("%d",&m[i]);
        for (int j=1;j<=m[i];j++)
        {
            int x,y;scanf("%d%d",&x,&y);
            e[i].push_back((hh){x,y});
        }
    }
    int tot=1<<n-1;LL ans=0;
    for (int i=0;i<tot;i++)
    {
        int id=0;memset(a,0,sizeof(a));
        for (int j=0;j<n;j++)
          if (i>>j&1)
          {
            id++;
            for (int k=1;k<=m[j+1];k++) 
            {
                hh now=e[j+1][k-1];
                int x=now.x,y=now.y;
                a[x][y]--; a[y][x]--;
                a[x][x]++; a[y][y]++;
            }
          }
        int flag=n-1-id;LL t=gauss(n-1);
        if (flag&1) ans-=t;else ans+=t;
        ans=(ans+mod)%mod;
    }
    printf("%lld",ans);
}