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;
}