BZOJ 4596 [SHOI2016]黑暗前的幻想乡

版权声明：本文为博主原创文章，未经博主允许必须转载。 https://blog.csdn.net/qq_35950004/article/details/88192515

题目
这个我觉得可以加强：求恰好由k个建筑公司施工的方案数。
广义容斥定理更能区分出选手是否真的会容斥还是只会个公式和小学奥数锻炼出来的直觉。
当练习 $O(n^3)$求行列式了。

#include<bits/stdc++.h>
#define maxn 18
#define maxm 400
#define mod 1000000007
using namespace std;

int n;
int c[maxn][maxn],a[maxn][maxn];
int m[maxn],x[maxn][maxm],y[maxn][maxm];

int solve()
{
	memcpy(a,c,sizeof c);
	int ret = 1;
	for(int i=2;i<=n;i++)
		for(int j=i+1;j<=n;j++)
			if(a[j][i])
			{
				if(!a[i][i])
				{
					for(int k=i;k<=n;k++)
						swap(a[i][k],a[j][k]);
					ret = -ret;
				}
				else 
				{
					int b[2][2]={{1,0},{0,1}},d[2]={a[i][i],a[j][i]};
					while(d[1])
					{
						swap(d[0],d[1]),swap(b[0][0],b[0][1]),swap(b[1][0],b[1][1]);
						ret = -ret;
						int tmp = d[1] / d[0];
						d[1] -= tmp * d[0];
						b[0][1] = (b[0][1] - 1ll * tmp * b[0][0]) % mod;
						b[1][1] = (b[1][1] - 1ll * tmp * b[1][0]) % mod;
					}
					
					for(int k=i;k<=n;k++)
					{
						int x = a[i][k] , y = a[j][k];
						a[i][k] = (1ll * b[0][0] * x + 1ll * b[1][0] * y) % mod;
						a[j][k] = (1ll * b[0][1] * x + 1ll * b[1][1] * y) % mod;
					}
				}
			}
	for(int i=2;i<=n;i++)
		ret = 1ll * ret * a[i][i] % mod;
	return ret;
}

int ans = 0;

void dfs(int now,int had)
{
	if(now == n)
	{
		ans = (ans + (((n-1-had)&1)?-1:1) * solve()) % mod;
		return;
	}
	dfs(now+1,had);
	for(int i=1;i<=m[now];i++)
		c[x[now][i]][y[now][i]] --,
		c[y[now][i]][x[now][i]] --,
		c[x[now][i]][x[now][i]] ++,
		c[y[now][i]][y[now][i]] ++;
	dfs(now+1,had+1);
	for(int i=1;i<=m[now];i++)
		c[x[now][i]][y[now][i]] ++,
		c[y[now][i]][x[now][i]] ++,
		c[x[now][i]][x[now][i]] --,
		c[y[now][i]][y[now][i]] --;
}

int main()
{
	scanf("%d",&n);
	for(int i=1;i<n;i++)
	{
		scanf("%d",&m[i]);
		for(int j=1;j<=m[i];j++)
			scanf("%d%d",&x[i][j],&y[i][j]);
	}
	dfs(1,0);
	printf("%d\n",(ans+mod)%mod);
}