NYOJ 27-水池数目(简单DFS)

题目链接：
http://acm.nyist.net/JudgeOnline/problem.php?pid=27

水池数目

时间限制：3000 ms  |  内存限制：65535 KB
难度：4

描述南阳理工学院校园里有一些小河和一些湖泊，现在，我们把它们通一看成水池，假设有一张我们学校的某处的地图，这个地图上仅标识了此处是否是水池，现在，你的任务来了，请用计算机算出该地图中共有几个水池。

输入第一行输入一个整数N，表示共有N组测试数据

每一组数据都是先输入该地图的行数m(0<m<100)与列数n(0<n<100)，然后，输入接下来的m行每行输入n个数，表示此处有水还是没水（1表示此处是水池，0表示此处是地面）
输出输出该地图中水池的个数。

要注意，每个水池的旁边（上下左右四个位置）如果还是水池的话的话，它们可以看做是同一个水池。

样例输入

2
3 4
1 0 0 0
0 0 1 1
1 1 1 0
5 5
1 1 1 1 0
0 0 1 0 1
0 0 0 0 0
1 1 1 0 0
0 0 1 1 1

样例输出

2
3

//典型的求连通块,DFS
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;

int s[105][105];
int vis[105][105];
int di[][2] = {{1,0},{-1,0},{0,1},{0,-1}};
int m, n;

int valid(int x, int y)
{
if(x<0||x>=m||y<0||y>=n)
return 0;
return 1;
}

void dfs(int x, int y)
{
vis[x][y] = 1;
for(int i=0; i<4; i++)
{
int xx = x+di[i][0];
int yy = y+di[i][1];
if(s[xx][yy]==1&&valid(xx, yy)&&!vis[xx][yy])
{
dfs(xx, yy);
}
}
}

int main()
{
int N;
scanf("%d", &N);
while(N--)
{
memset(vis, 0, sizeof(vis));
scanf("%d%d", &m, &n);
for(int i=0; i<m; i++)
for(int j=0; j<n; j++)
scanf("%d", &s[i][j]);
int ans = 0;
for(int i=0; i<m; i++)
for(int j=0; j<n; j++)
if(!vis[i][j]&&s[i][j]==1)
{
dfs(i, j);
ans++;
}
printf("%d\n", ans);
}
return 0;
}