nyoj 27 水池数目【DFS】

水池数目

时间限制： 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

解题方法：深度优先搜索

# include <stdio.h>
# include <string.h>
# define M 105

int color[M][M];//水池是否被染色（即是否遍历过）
int poolnum;    //水池的数目
int r, c;
int map[M][M];//存放地图

void dfs(int i, int j);

int main(void)
{
	int n;
	scanf("%d", &n);
	while (n--)
	{
		int i, j;
		poolnum = 0;
		memset(color, 0, sizeof(color));
		scanf("%d %d", &r, &c);
		for (i = 0; i < r; i++)
		{
			for (j = 0; j < c; j++)
			{
				scanf("%d", &map[i][j]);
			}
		}
		for (i = 0; i < r; i++)
		{
			for (j = 0; j < c; j++)
			{
				if (!color[i][j] && map[i][j])
				{
					poolnum++;
					dfs(i, j);
				}
			}
		}
		printf("%d\n", poolnum);
	
	}
	return 0;
}

void dfs(int i, int j)
{
	if (color[i][j])
	{
		return ;
	}
	int next[4][2] = {{0, 1}, {1, 0}, {0, -1}, {-1, 0}};//方向变量
        color[i][j] = poolnum;
	for (int k = 0; k < 4; k++)
	{
		int tx = i + next[k][0];
		int ty = j + next[k][1];
		if (tx >= r || tx < 0 || ty >= c || ty < 0)
		{
			continue;
		}
		if (map[tx][ty])
		{
			
			dfs(tx, ty);
		}
	}
}