hdu1198Farm Irrigation -并查集

思路：求无向图的联通分量个数
方法：1.对于从A,到K四个正方形，每一条边的中点若有水管则为1，没有水管则为0，按照逆时针方向从上，左，下，右依次排序
例如：正方形A可以表示为{1， 1， 0， 0}
2. 我们只要判断每一个小正方形的四条边的中点是不是等于其左侧和上侧的正方形的中点，并且同时为1,若相等则归为一类，
   此时我们要以给每一个小正方形一个编号 i * n + j 其中（i,j）表示第i行的第j个小正方形

3. 最后遍历这个m * n这个小正方形，计算书parent[i] = i的个数

#include <iostream>
using namespace std;

#define MSIZ 3000
#define N 60
int graph[N][N];
int parent[MSIZ];
int m_rank[MSIZ];
// 上，左，下，右
int arg[11][4] = {
	{1, 1, 0, 0},
	{1, 0, 0, 1},
	{0, 1, 1, 0},
	{0, 0, 1, 1},
	{1, 0, 1, 0},
	{0, 1, 0, 1},
	{1, 1, 0, 1},
	{1, 1, 1, 0},
	{0, 1, 1, 1},
	{1, 0, 1, 1},
	{1, 1, 1, 1}
};
int m, n;

void init()
{
	int i = 0;
	for(i = 0;i < MSIZ; ++i)
	{
		parent[i] = i;
		m_rank[i] = 1;
	}
}

int findSet(int x)
{
	if (x != parent[x])
	{
		parent[x] = findSet(parent[x]);
	}
	return parent[x];
}

void unionSet(int x,int y)
{
	x = findSet(x);
	y = findSet(y);

	if (x != y)
	{
		parent[x] = y;
		m_rank[y] += m_rank[x];
	}

}

int main()
{
	int i, j;
	int x, y;
	char tmp;
	int num = 0;
	while(scanf("%d %d", &m, &n) != EOF )
	{
		getchar();
		if( m == -1 && n == -1)
		{
			break;
		}
		init();
		
		
		num = 0;

		for (i = 0;i < m; ++i)
		{
			for (j = 0; j < n; ++j)	
			{
				scanf("%c ", &tmp);
				
				graph[i][j] = tmp - 'A';
				if (i == 0 && j == 0 )
				{
					continue;
				}

				if (i >= 1 && arg[graph[i][j]][0] == 1 && arg[graph[i][j]][0] == arg[graph[i - 1][j]][2])
				{
					x = i * n + j;
					y = (i - 1)* n + j;
					unionSet(x, y);
				}

				if (j >= 1 && arg[graph[i][j]][1] == 1 && arg[graph[i][j]][1] == arg[graph[i][j-1]][3])
				{
					x = i * n + j;
					y = i * n + j - 1;
				
					unionSet(x, y);

				}

			}
		}
		int total = m * n;
		for (i = 0;i < total ; ++i)
		{
			if (parent[i] == i)
			{
				num++;
			}
		}
		printf("%d\n", num);
	}

	return 0;
}