hdu 1198 dfs水(连通性)

点击打开链接

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
const int M =60;
int g[M][M];// g[i][j] 中存的是PIPE的种类 
typedef struct{
	int x;
	int y;
}move1;
move1 Move[4]={{-1,0},{1,0},{0,-1},{0,1}};
// 0 1 2 3  up  down left right
// arr[i][j]  两个Pipe 能相连当且仅当 arr[x][j1] arr[y][j2]为1 
int arr[11][4]={{1,0,1,0},{1,0,0,1},{0,1,1,0},{0,1,0,1},{1,1,0,0}
			,{0,0,1,1},{1,0,1,1},{1,1,1,0},{0,1,1,1},{1,1,0,1},{1,1,1,1}};


int n,m,visit[M][M];		
void dfs(int x,int y)
{
	
	for(int i=0;i<4;i++)
	{
		int a=x+Move[i].x;
		int b=y+Move[i].y; 
		if(a>=1&&a<=n&&b>=1&&b<=m) 
		{
			int j;
			switch(i) // 往i方向到(a,b)时  （a,b) 应该有 j方向的pipe 
			{
				case 0:j=1; break;
				case 1:j=0; break;
				case 2:j=3; break;
				case 3:j=2; break;	
			} 
			if(arr[g[x][y]][i]==1&&arr[g[a][b]][j]==1&&!visit[a][b])
			{
				visit[a][b]=1; 
				dfs(a,b);
			
			
			}
			
		}
	}
}


int main()
{
	while(cin>>n>>m&&m>0&&n>0)
	{
		memset(visit,0,sizeof(visit));
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=m;j++)
			{
				char c;
				cin>>c;
				g[i][j]=c-'A';
			}	
		} 
		int ans=0;
		// 在同一个连通分量中 只要装一个Wellsprings即可 装在那个地方效果都一样 
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=m;j++)
			{
				if(!visit[i][j])
				{
				
					ans++;
					visit[i][j]=1; 
					dfs(i,j); //遍历该连通分量 
				
				}
			}
		}
		cout<<ans<<endl;
	}
	return 0;
}