BZOJ4596 黑暗前的幻想乡 Matrix_Tree定理 容斥原理

题意：n个点的图其中每条边都有颜色且共有n-1种颜色，求一个生成树满足每种颜色的边出现且仅出现一次

Sol:

容斥全靠猜系列

答案=所有边生成树个数-少一种颜色生成树个数+少两个...

生成树个数用矩阵树定理算一下，注意符号

Code:

#include<bits/stdc++.h>
typedef long long ll;
using namespace std;
const int maxn = 19;
const int mod = 1e9+7;
const double eps = 1e-10;
 
struct Item
{
	int sz;
	vector<pair<int,int> > line;
}a[maxn];
ll f[maxn][maxn];
int n,base,ans;
 
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
 
ll PowerMod(ll x,int y)
{
	ll res=1;
	while(y)
	{
		if(y&1) res=res*x%mod;
		x=x*x%mod;y>>=1;
	}return res;
}
 
inline int count(int x)
{
	int res=0;
	while(x){if(x&1)res++;x>>=1;}
	return res;
}
 
ll gauss(int len)
{
	for(int i=1;i<=len;i++)
		for(int j=1;j<=len;j++)
			f[i][j]=(f[i][j]+mod)%mod;
	ll res=1;int opt=1;
	for(int i=1;i<=len;i++)
	{
		int now=i;
		while(now<=len&&!f[now][i]) now++;
		if(now>len) return 0;
		if(now!=i)
		{
			for(int j=1;j<=len;j++) swap(f[now][j],f[i][j]);
			opt=-opt;
		}
		ll tmp=f[i][i],inv=PowerMod(tmp,mod-2);
		for(int j=1;j<=len;j++) f[i][j]=f[i][j]*inv%mod;
		res=res*tmp%mod;
		for(int j=i+1;j<=len;j++)
		{
			tmp=f[j][i];
			for(int k=1;k<=len;k++) f[j][k]=(mod+f[j][k]-f[i][k]*tmp%mod)%mod;
		}
	}
	return (opt*res+mod)%mod;
}
 
void ClearLove(){memset(f,0,sizeof f);}
int solve(int sta)
{
	ClearLove();
	for(int i=1;i<n;i++) if(sta&(1<<i>>1))
	{
		for(int j=0;j<a[i].sz;j++)
		{
			int x=a[i].line[j].first,y=a[i].line[j].second;
			f[x][y]--;f[y][x]--;f[x][x]++;f[y][y]++;
		}
	}
	int res=gauss(n-1);
	return res;
}
 
int main()
{
	n=read();
	for(int i=1;i<n;i++)
	{
		a[i].sz=read();
		for(int j=1;j<=a[i].sz;j++)
			a[i].line.push_back(make_pair(read(),read()));
	}
	int mr=(1<<(n-1))-1;base=count(mr);
	for(int i=1;i<=mr;i++)
	{
		int opt=((count(i)&1)==(base&1))?1:-1;
		ans=(ans+opt*solve(i))%mod;
	}
	cout<<(ans+mod)%mod<<endl;
	return 0;
}