水池数目

水池数目

时间限制：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 <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <cstdlib>
using namespace std;
#define mem(a) memset(a, 0, sizeof(a))
int a[110][110], vis[110][110], n, m;
int dx[4] = {1, 0, -1, 0}, dy[4] = {0, 1, 0, -1};

int check(int i, int j) {
	if (!vis[i][j] && i>=1 && i<=n && j>=1 && j<=m)	return 1;
	return 0;
}

void dfs(int x, int y) {
	a[x][y] = 0;
	vis[x][y] = 1;
	for (int i = 0; i<4; i++) {
		int nx = x + dx[i], ny = y + dy[i];
		if (check(nx, ny) && a[nx][ny] == 1) 	dfs(nx, ny);
	}
}

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