bzoj 4596: [Shoi2016]黑暗前的幻想乡【容斥原理+矩阵树定理】

真是简单粗暴
把矩阵树定理的运算当成黑箱好了反正我不会
这样我们就可以在O(n^3)的时间内算出一个无向图的生成树个数了
然后题目要求每个工程队选一条路,这里可以考虑容斥原理:全选的方案数-不选工程队1能修的路的方案数-不选工程队2能修的路的方案数……+不选工程队12能修的路的方案数+不选工程队13能修的路的方案数……-不选工程队123能修的路的方案数……
这里直接O(2^(n-1))枚举选择状态即可,然后根据不选的个数奇偶来决定在ans上减或加即可

#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>
using namespace std;
const int N=20,mod=1e9+7;
int n,a[N][N];
long long ans;
vector<pair<int,int> >v[N];
int read()
{
	int r=0,f=1;
	char p=getchar();
	while(p>'9'||p<'0')
	{
		if(p=='-')
			f=-1;
		p=getchar();
	}
	while(p>='0'&&p<='9')
	{
		r=r*10+p-48;
		p=getchar();
	}
	return r*f;
}
int gaosi(int n)
{
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			if(a[i][j]<0)
				a[i][j]+=mod;
	long long ans=1,f=1;
	for(int i=1;i<=n;i++)
	{
		for(int j=i+1;j<=n;j++)
		{
			long long x=a[i][i],y=a[j][i];
			while(y)
			{
				long long t=x/y;
				x%=y;
				swap(x,y);
				for(int k=i;k<=n;k++)
					a[i][k]=(a[i][k]-a[j][k]*t%mod+mod)%mod;
				for(int k=i;k<=n;k++)
					swap(a[i][k],a[j][k]);
				f=-f;
			}
		}
		if(a[i][i]==0)
			return 0;
		ans=ans*a[i][i]%mod;
	}
	return f==-1?(mod-ans)%mod:ans;
}
int main()
{
	n=read();
	for(int i=1;i<n;i++)
	{
		int m=read();
		for(int j=1;j<=m;j++)
		{
			int x=read(),y=read();
			v[i].push_back(make_pair(x,y));
		}
	}
	for(int i=0;i<(1<<(n-1));i++)
	{
		int s=n-1,x=i;
		memset(a,0,sizeof(a));
		for(int j=1;j<n;j++,x>>=1)
			if(x&1)
			{
				s--;
				for(int k=0;k<v[j].size();k++)
					a[v[j][k].first][v[j][k].first]++,a[v[j][k].second][v[j][k].second]++,a[v[j][k].first][v[j][k].second]--,a[v[j][k].second][v[j][k].first]--;
			}
		if(s&1)
			ans=(ans-gaosi(n-1))%mod;
		else
			ans=(ans+gaosi(n-1))%mod;
	}
	printf("%d\n",(ans+mod)%mod);
	return 0;
}
posted @ 2018-07-02 11:54  lokiii  阅读(161)  评论(0编辑  收藏  举报