HDOJ1198

简单的并查集，很适合刚学并查集拿来练手，t贴一下源码

#include<cstdio>
#include<set>
#include<cstring>
using namespace std;

/*HDU1198 并查集*/
struct Node{
	int l,r,t,b;//1 or 0
	Node(int a,int bb,int c,int d){l=a;r=bb;t=c;b=d;}
	Node(){}
};
Node num[]={Node(1,0,1,0),Node(0,1,1,0),Node(1,0,0,1),Node(0,1,0,1),Node(0,0,1,1),Node(1,1,0,0),
			  Node(1,1,1,0),Node(1,0,1,1),Node(1,1,0,1),Node(0,1,1,1),Node(1,1,1,1)};
const int maxn=55;
int m,n;
Node nodes[maxn][maxn];
int par[maxn*maxn],hg[maxn*maxn];
void init(int n){
	for(int i=0;i<n;i++){
		par[i]=i;
		hg[i]=0;	
	}
	
}
int find(int x){
	if(par[x]==x) return x;
	return par[x]=find(par[x]);
}
void unite(int x,int y){
	x=find(x),y=find(y);
	if(x==y) return ;
	if(hg[x]<hg[y]) par[x]=y;
	else{
		par[y]=x;
		if(hg[x]==hg[y]) hg[x]++;
	}
}
int solve(){
	init(m*n);
	//遍历，每一个都考虑右下
	for(int i=0;i<m;i++){
		for(int j=0;j<n;j++){
			if(j+1<n){
				if(nodes[i][j].r&&nodes[i][j+1].l) unite(i*n+j,i*n+j+1);
			}
			if(i+1<m){
				if(nodes[i][j].b&&nodes[i+1][j].t) unite(i*n+j,(i+1)*n+j);	
			}
		}
	}
	set<int> s;
	for(int i=0;i<n*m;i++)
		s.insert(find(i));  //!!!!注意这里不要写成par[i]
	return s.size();
}
int main()
{
	freopen("data.txt","r",stdin);
	scanf("%d %d\n",&m,&n);
	while(m>0&&n>0){
		for(int i=0;i<m;i++){
			char c;
			for(int j=0;j<n;j++){
				scanf("%c",&c);
				nodes[i][j]=num[c-'A'];
			}
			scanf("%c",&c);
		}
		int ans=solve();
		//int ans=2;
		printf("%d\n", ans);
		scanf("%d %d\n",&m,&n);
	}
	return 0;
}