HDU 1198 Farm Irrigation

传送门

这个题就是考建图，其实想起来以前玩的一个手机游戏就是这种类似铺管道的。
在矩阵里，每个点往4个方向看一下能不能和相邻点接上，但是这样的话会重复添加边（19.5.8 更新：其实重复添也没关系，但是完全没必要。。）。更好的想法是往右下两个方向就行了，遍历矩阵按照外层从上到下，内层从左到右。

---

（19年5月8日）他问你这个图最少布置多少个泉眼（wellspring），从而可以灌溉到整个田地。其实就是问你这个无向图中有多少个连通分量。

对于无向无环图，有 连通分量数 = 点数 - 边数 成立（这里统计的边数都不能构成环！用并查集添的边符合这个性质）。可以想到，若原图有环，破坏掉所有的环并不会改变原有的连通分量数。

为什么成立？因为对于 无向无环连通图 或 无向无环连通分量，满足：点数 - 边数 = 1。可参考HDU 1272

所以，虽然这道题的无向图可能有环，但是只需要用并查集看看最多能添多少边就可以了（上限是点数-1），因为这道题解决的是 有几个连通分量 / 无向图的连通性，有没有环并不影响这个问题的答案，所以实际上是用并查集当作无环图求解。

#include <iostream>
#include <algorithm>
#include <vector>
#include <cstring>
using namespace std;

const int MAXN = 50;
int M, N;
char mtx[MAXN][MAXN];

int dx[2] = { 0,1 };   // 右下
int dy[2] = { 1,0 };
bool u[11] = { 0 };
bool l[11] = { 0 };    // 不能用 left 和 right 作为变量名
bool d[11] = { 0 };
bool r[11] = { 0 };

int pre[MAXN*MAXN];
int conn;

struct Edge
{
	int n1, n2;
};
vector<Edge> E;

void create_graph()
{
	for (int i = 0; i < M; i++)
	{
		for (int j = 0; j < N; j++)
		{
			char c = mtx[i][j];
			for (int k = 0; k < 2; k++)    // 从(0,0)开始，其实只用向右下两个方向跑就行   
			{
				int i0 = i + dx[k];
				int j0 = j + dy[k];
				if (i0 < M && i0 >= 0 && j0 < N && j0 >= 0)
				{
					char c0 = mtx[i0][j0];
					if (k == 0 && r[c - 'A'] && l[c0 - 'A'] || k == 1 && d[c - 'A'] && u[c0 - 'A'])
						E.push_back(Edge{ i*N + j,i0*N + j0 });
				}
			}
		}
	}
}

void init_char()
{
	u[0] = 1; l[0] = 1;
	u[1] = 1; r[1] = 1;
	l[2] = 1; d[2] = 1;
	r[3] = 1; d[3] = 1;
	u[4] = 1; d[4] = 1;
	l[5] = 1; r[5] = 1;
	u[6] = 1; l[6] = 1; r[6] = 1;
	u[7] = 1; d[7] = 1; l[7] = 1;
	l[8] = 1; r[8] = 1; d[8] = 1;
	u[9] = 1; d[9] = 1; r[9] = 1;
	u[10] = 1; l[10] = 1; d[10] = 1; r[10] = 1;
}

void init()
{
	memset(pre, -1, sizeof pre);
	conn = 0;
	E.clear();
}

int f(int n)
{
	if (pre[n] < 0) return n;
	return pre[n] = f(pre[n]);
}

void un(int n1, int n2)
{
	int f1 = f(n1);
	int f2 = f(n2);
	if (f1 != f2)
	{
		conn++;
		if (pre[f1] <= pre[f2])
		{
			pre[f1] += pre[f2];
			pre[f2] = f1;
		}
		else
		{
			pre[f2] += pre[f1];
			pre[f1] = f2;
		}
	}
}

int main()
{
	init_char();
	for (; ~scanf("%d%d", &M, &N);)
	{
		if (M < 0 || N < 0) break;
		init();
		for (int i = 0; i < M; i++)
		{
			getchar();                     // 换行符
			for (int j = 0; j < N; j++)
				scanf("%c", &mtx[i][j]);
		}

		create_graph();
		for (int i = 0; i < E.size(); i++)
		{
			if (conn == M*N - 1) break;    // 可以早点退出了，再运行下去conn也不可能再增加了
			un(E[i].n1, E[i].n2);
		}
		// 没删调试信息，调错1个小时。
		printf("%d\n", M*N - conn);
	}

	return 0;
}