NYOJ - 27：水池数目

水池数目

来源：NYOJ

标签：图论，深度优先搜索

参考资料：

相似题目：

题目

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

输入

第一行输入一个整数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

解题思路

深度优先搜索的次数即是水池数目。

参考代码

#include
#include
#define MAXN 105

int dx[4]={0,-1,0,1};
int dy[4]={-1,0,1,0};

int map[MAXN][MAXN];
int m,n;

void dfs(int x,int y){
	map[x][y]=-1;
	for(int i=0;i<4;i++){
		int nx=x+dx[i];
		int ny=y+dy[i];
		if(nx>=0 && nx<m && ny>=0 && ny<=n && map[nx][ny]==1){
			map[nx][ny]==-1;
			dfs(nx,ny);
		}
	}
}

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