【ybt金牌导航8-2-4】【luogu P4111】你的房间 / 小 Z 的房间(矩阵树定理)(高斯消元)

你的房间 / 小 Z 的房间

题目链接:ybt金牌导航8-2-4 / luogu P4111

题目大意

给你一个二维网格,然后有一些位置是房间,然后房间和房间之间的道路你可以打通。
然后问你有多少种打通方法使得任意两个房间都有且只有一条道路。

思路

其实你会发现有且只有一条道路的话就是要把它弄成类似树的形式。

然后就变成了类似生成树计数之类的?
自然想到矩阵树定理,那我们就把边权设成是无向边,边权为 \(1\) 即可。

代码

#include<cstdio>
#include<algorithm>
#define ll long long
#define mo 1000000000

using namespace std;

int n, m, tot, bh[101][101];
int dx[4] = {1, 0, -1, 0}, dy[4] = {0, 1, 0, -1};
bool a[11][11];
char c;
ll f[101][101];

bool ck(int x, int y) {
	if (x < 1 || x > n) return 0;
	if (y < 1 || y > m) return 0;
	return 1;
}

int get_bh(int x, int y) {
	if (!bh[x][y]) bh[x][y] = ++tot;
	return bh[x][y];
}

ll work(int l, int r) {
	ll ans = 1, zf = 1, tmp;
	for (int i = l; i <= r; i++) {
		int k = i;
		for (int j = i + 1; j <= r; j++)
			if (f[j][i] > f[k][i]) k = j;
		if (k != i) zf = -zf, swap(f[i], f[k]);
		if (!f[i][i]) return 0;
		for (int j = i + 1; j <= r; j++) {
			if (f[j][i] > f[i][i]) zf = -zf, swap(f[i], f[j]);
			while (f[j][i]) {
				tmp = f[i][i] / f[j][i];
				for (int k = i; k <= r; k++)
					f[i][k] = (f[i][k] + f[j][k] * (mo - tmp) % mo) % mo;
				swap(f[i], f[j]); zf = -zf;
			}
		}
		ans = ans * f[i][i] % mo;
	}
	if (zf == -1) return (mo - ans) % mo;
		else return ans;
}

int main() {
	scanf("%d %d", &n, &m);
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= m; j++) {
			c = getchar();
			while (c != '.' && c != '*') c = getchar();
			if (c == '.') a[i][j] = 1;
		}
	
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= m; j++)
			if (a[i][j]) get_bh(i, j);
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= m; j++) {
			if (!a[i][j]) continue;
			for (int k = 0; k < 4; k++) {
				int x = i + dx[k], y = j + dy[k];
				if (!ck(x, y)) continue;
				if (!a[x][y]) continue;
				f[get_bh(i, j)][get_bh(x, y)] = (f[get_bh(i, j)][get_bh(x, y)] - 1 + mo) % mo;
				f[get_bh(x, y)][get_bh(x, y)] = (f[get_bh(x, y)][get_bh(x, y)] + 1) % mo;
			}
		}
	
	if (tot == 1) {
		printf("1"); return 0;
	}
	printf("%lld", work(2, tot));
	
	return 0;
}
posted @ 2021-12-08 08:57  あおいSakura  阅读(31)  评论(0编辑  收藏  举报