[洛谷P4336] SHOI2016 黑暗前的幻想乡

问题背景

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

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

问题描述

幽香上台以后，第一项措施就是要修建幻想乡的公路。幻想乡一共有 n 个城市，之前原来没有任何路。幽香向选民承诺要减税，所以她打算只修 n-1 条公路将这些城市连接起来。但是幻想乡有正好 n-1 个建筑公司，每个建筑公司都想在修路地过程中获得一些好处。虽然这些建筑公司在选举前没有给幽香钱，幽香还是打算和他们搞好关系，因为她还指望他们帮她建墙。所以她打算让每个建筑公司都负责一条路来修。

每个建筑公司都告诉了幽香自己有能力负责修建的路是哪些城市之间的。所以幽香打算 n - 1 条能够连接幻想乡所有城市的边，然后每条边都交给一个能够负责该边的建筑公司修建，并且每个建筑公司都恰好修建一条边。

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

输入格式

第一行包含一个整数 n，表示城市个数。

第 2 到第 n 行，第 (i + 1) 行表示 第 i 个建筑公司可以修建的路的列表：以一个非负整数 mi 开头，表示其可以修建条路的条数；接下来有 mi 对整数 u, v，每对数表示一条边的两个端点。其中不会出现重复的边，也不会出现自环。

输出格式

输出一行一个整数，表示所有可能的方案数对 \(10^9+7\) 取模的结果。

样例输入

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

样例输出

17

数据范围

对于\(100\%\) 的测试点，\(2 \leq n \le 17\)，\(0 \leq m_i \leq \frac{n(n - 1)}{2}\)，\(1 \leq u, v \leq n\)。

解析

所有建筑公司能够修建的道路构成了一张图，而最后修建的道路一定构成了原图的一个生成树。\(n\) 的范围如此的小，我们不妨考虑容斥。枚举哪些建筑公司没有可修建的道路，并将枚举的建筑公司的边集从原图中删去。那么剩下的图中若还能构成生成树，说明这棵生成树一定是不合法的。我们可以对剩下的图也统计一次生成树个数。结合容斥原理，设 \(f_S\)表示至少集合 \(S\) 中的公司没有修路时的方案数：

\[Ans=\sum_{S\in \{1...n\}}(-1)^{|S|}f_{S} \]

生成树的方案数可以用矩阵树定理计算。

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#define int long long
#define N 20
using namespace std;
const int mod=1000000007;
struct edge{
	int u,v;
}e[N][N*N];
int n,i,j,m[N],ans;
int g[N][N],d[N][N];
bool vis[N];
int read()
{
	char c=getchar();
	int w=0;
	while(c<'0'||c>'9') c=getchar();
	while(c<='9'&&c>='0'){
		w=w*10+c-'0';
		c=getchar();
	}
	return w;
}
int poww(int a,int b)
{
	int ans=1,base=a;
	while(b){
		if(b&1) ans=ans*base%mod;
		base=base*base%mod;
		b>>=1;
	}
	return ans;
}
int gcd(int a,int b)
{
	if(b==0) return a;
	return gcd(b,a%b);
}
int solve()
{
	int ans=1,dv=1;
	for (int i=1, j=1; i<n && j<n; ++i, ++j)
	{
		int id=i;
		for (int k=i; k<n; ++k)
			if (d[k][j]!=0) { id=k; break; }
		if (d[id][j]==0) { --i; continue; }
		if (id!=i)
		{
			ans*=-1;
			for (int k=j; k<n; ++k) swap(d[id][k], d[i][k]);
		}
		for (int k=i+1; k<n; ++k)
			if (d[k][j]!=0)
			{
				int tmp1=gcd(abs(d[i][j]), abs(d[k][j]));
				int tmp2=d[i][j]/tmp1;
				tmp1=d[k][j]/tmp1;
				dv=dv*tmp2%mod;
				for (int p=j; p<n; ++p)
					d[k][p]=(d[k][p]*tmp2-d[i][p]*tmp1+mod)%mod;
			}
	}
	for(int i=1;i<n;i++) ans=(ans*d[i][i]%mod+mod)%mod;
	return (ans*poww(dv,mod-2)%mod+mod)%mod;
}
void dfs(int x)
{
	if(x==n){
		memset(g,0,sizeof(g));
		memset(d,0,sizeof(d));
		int cnt=0;
		for(int i=1;i<n;i++){
			if(!vis[i]){
				for(int j=1;j<=m[i];j++){
					int u=e[i][j].u,v=e[i][j].v;
					g[u][v]++;g[v][u]++;
					d[u][u]++;d[v][v]++;
				}
			}
			else cnt++;
		}
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++) d[i][j]-=g[i][j];
		}
		if(cnt%2==0) ans=(ans+solve())%mod;
		else ans=(ans-solve()+mod)%mod;
		return;
	}
	dfs(x+1);
	vis[x]=1;dfs(x+1);vis[x]=0;
}
signed main()
{
	n=read();
	for(i=1;i<n;i++){
		m[i]=read();
		for(j=1;j<=m[i];j++) e[i][j].u=read(),e[i][j].v=read();
	}
	dfs(1);
	printf("%lld\n",ans);
	return 0;
}

标签：幻想,洛谷,int,幽香,修建,ans,P4336,SHOI2016,建筑
来源： https://www.cnblogs.com/LSlzf/p/13382669.html