nyoj27 水池数目

水池数目

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

点击此处原题链接

思路：找到有水点向四周扩散，经过的都标记为0,遍历的次数即为水池的个数

//广搜（bfs）代码： 
#include <stdio.h>
#include <string.h>
#include <queue>
using namespace std;
int n,m;
int a[110][110];
int count ;
int der[4][2] = {{0,1},{0,-1},{1,0},{-1,0}};
struct Node
{
	int x,y;
};
queue<Node> Q;
Node p,q;
void bfs(int x,int y)
{
	a[x][y] = 0;
	p.x = x;
	p.y = y;
	Q.push(p);
	while(!Q.empty())
	{
		q = Q.front();
		for(int i = 0;i < 4;i ++ )
		{
			if(a[q.x + der[i][0]][q.y + der[i][1]])
			{
				a[q.x + der[i][0]][q.y + der[i][1]] = 0;
				p.x = q.x + der[i][0];
				p.y = q.y + der[i][1];
				Q.push(p);
			}
		}
		Q.pop();
	}
}
int main()
{
	int N;
	scanf("%d",&N);
	while(N -- )
	{
		int i,j;
		memset(a,0,sizeof(a));
		scanf("%d%d",&m,&n);
		for(i = 1;i <= m;i ++ )
		{
			for(j = 1;j <= n;j ++ )
				scanf("%d",&a[i][j]);
		}
		count = 0;
		for(i = 1;i <= m;i ++ )
		{
			for(j = 1;j <= n;j ++ )
				if(a[i][j] == 1)
				{
					bfs(i,j);
					count ++ ;
				}
		}
		printf("%d\n",count);
	}
	return 0;
}

//深搜（dfs）代码： 
#include <stdio.h>
#include <string.h>
int map[110][110];
int m,n;
int ans;
void dfs(int x,int y)
{
	map[x][y] = 0;
	if( x > 0 && map[x - 1][y] == 1)		dfs(x - 1,y);
	if( x < m - 1 && map[x + 1][y] == 1)	dfs(x + 1,y);
	if( y > 0 && map[x][y - 1] == 1)		dfs(x,y - 1);
	if( y < n - 1 && map[x][y + 1] == 1)	dfs(x,y + 1);
}
int main()
{
	int T;
	scanf("%d",&T);
	while(T -- )
	{	
		scanf("%d%d",&m,&n);
		int i,j;
		for(i = 0;i < m;i ++ )
		{
			for(j = 0;j < n;j ++ )
			scanf("%d",&map[i][j]);
		}
		ans = 0;
		for(i = 0;i < m;i ++ )
		{
			for(j = 0;j < n;j ++ )
			if(map[i][j] == 1)
			{
				dfs(i,j);
				ans ++ ;
			}
		}
		printf("%d\n",ans);
	}
	return 0;
}