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