hdu 1198 Farm Irrigation(并查集)

http://acm.hdu.edu.cn/showproblem.php?pid=1198

题目大意：Benny将农场分割成一些小矩形，每个矩形都放一种形状的水管，问需要多少水源能浇灌整个农场。

                    将每种形状的水管四个方向标记，能流通为1，不嗯能够流通为0。

                    之后就可以讲流通的水管并起来。

代码如下：

#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<string>
#include<iostream>
#include<algorithm>
using namespace std;
int set[3000],map[][4]={1,1,0,0,0,1,1,0,1,0,0,1,0,0,1,1,0,1,0,1,1,0,1,0,1,1,1,0,1,1,0,1,1,0,1,1,0,1,1,1,1,1,1,1};//从A到K顺序保存四个方向
int find(int a)//找根节点
{
	while(a!=set[a])
		a=set[a];
	return a;
}
void Union(int a,int b)//合并
{
	a=find(a);
	b=find(b);
	set[a]=b;
}
bool zuoyou(int a,int b)//判断能否流通
{
	if(map[a][2]==1&&map[b][0]==1)
		return 1;
	return 0;
}
bool shangxia(int a,int b)//判断能否流通
{
	if(map[a][3]==1&&map[b][1]==1)
		return 1;
	return 0;
}
int main()
{
	int m,n,i,j,sum;
	char type[51][51];
	while(~scanf("%d%d%*c",&m,&n))
	{
		if(m==-1&&n==-1)
			break;
		for(i=1;i<=m*n;i++)
			set[i]=i;
		for(i=1;i<=m;i++)
		{
			for(j=1;j<=n;j++)
			{
				scanf("%c",&type[i][j]);
				if(i==1)
				{
					//以左上角为起点向右下角合并
					if(j!=1)
					{
						if(zuoyou(type[i][j-1]-'A',type[i][j]-'A')==1)//第一行的都只向左并
							Union((i-1)*n+j,(i-1)*n+j-1);
					}
				}
				else
				{
					if(j==1)
					{
						if(shangxia(type[i-1][j]-'A',type[i][j]-'A')==1)//第一列只向上并
							Union((i-2)*n+j,(i-1)*n+j);
					}
					else
					{
						//中间的都向左上并
						if(zuoyou(type[i][j-1]-'A',type[i][j]-'A')==1)
							Union((i-1)*n+j,(i-1)*n+j-1);
						if(shangxia(type[i-1][j]-'A',type[i][j]-'A')==1)
							Union((i-2)*n+j,(i-1)*n+j);
					}
				}
			}
			scanf("%*c");
		}
		sum=0;
		for(i=1;i<=m*n;i++)//查找根节点数
			if(set[i]==i)
				sum++;
			printf("%d\n",sum);
	}
	return 0;
}