BZOJ4596: [Shoi2016]黑暗前的幻想乡【矩阵树定理】

最新推荐文章于 2020-02-03 19:10:00 发布

原创最新推荐文章于 2020-02-03 19:10:00 发布 · 144 阅读

0 ·

CC 4.0 BY-SA版权

BZOJ 同时被 2 个专栏收录

203 篇文章

订阅专栏

矩阵树定理

3 篇文章

订阅专栏

博客围绕4596: [Shoi2016]黑暗前的幻想乡问题展开，提到考虑用矩阵树定理，但该定理无法解决选择n - 1个公司的问题，于是采用容斥原理，即总方案数减去一个公司不选的情况，加上两个公司不选的情况等。

4596: [Shoi2016]黑暗前的幻想乡

考虑矩阵树定理，但是不能解决十分选择了n-1个公司，所以我们可以容斥，总方案数-一个公司不选+两个公司不选-…

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MOD=1e9+7;
int n,cnt[1<<17],Ans,f[20][20];
struct Edge{int x,y;};
struct xcw{
	Edge que[200];int tot;
	void read(){++tot,scanf("%d%d",&que[tot].x,&que[tot].y);}
}a[20];
int Gauss(){
	int Now=1;
	for(int i=1;i<n;i++){
		for(int j=i+1;j<n;j++)
		while(f[j][i]){
			int t=f[i][i]/f[j][i];
			for(int k=i;k<n;k++) f[i][k]=(f[i][k]-1ll*t*f[j][k]%MOD+MOD)%MOD;
			swap(f[i],f[j]),Now=-Now;
		}
		Now=1ll*Now*f[i][i]%MOD;
	}
	return (Now+MOD)%MOD;
}
void Add(int x,int y){f[x][x]++,f[x][y]--,f[y][y]++,f[y][x]--;}
void Work(int x,int opt){
	memset(f,0,sizeof(f));
	for(int i=0;i<n-1;i++)
	if((x>>i)&1) for(int j=1;j<=a[i].tot;j++) Add(a[i].que[j].x,a[i].que[j].y);
	Ans=((Ans+opt*Gauss())%MOD+MOD)%MOD;
}
int main(){
	scanf("%d",&n);
	for(int i=0;i<n-1;i++){
		int K;scanf("%d",&K);
		for(int j=1;j<=K;j++) a[i].read();
	}
	for(int i=1;i<(1<<n-1);i++) cnt[i]=cnt[i>>1]+(i&1);
	for(int i=1;i<(1<<n-1);i++) Work(i,((n-cnt[i])&1)?1:-1);
	printf("%d\n",Ans);
	return 0;
}