南阳oj 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

入门级别的搜索，可用dfs也可用bfs，不多说贴代码

dfs:

#include<stdio.h>
#include<string.h>
int map[101][101];
int sum;
int n,m;
int move[4][2]={1,0,-1,0,0,1,0,-1};
void dfs(int x,int y)
{
map[x][y]=0;
for(int i=0;i<4;i++)
{
int xx=x+move[i][0];
int yy=y+move[i][1];
if(xx<0||xx>=n||yy<0||yy>=m||!map[xx][yy])
continue;
dfs(xx,yy);
}
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int i,j;
scanf("%d%d",&n,&m);
for(i=0;i<n;i++)
for(j=0;j<m;j++)
scanf("%d",&map[i][j]);
sum=0;
for(i=0;i<n;i++)
for(j=0;j<m;j++)
if(map[i][j])
{
sum++;
dfs(i,j);
}
printf("%d\n",sum);
}
}bfs:
#include<cstdio>
#include<queue>
using namespace std;
struct node
{
int x,y;
}no,ne;
int map[100][100];
int move[4][2]={1,0,-1,0,0,1,0,-1};
int n,m;
int sum;
void bfs(int x,int y)
{
int i;
queue<node>q;
no.x=x,no.y=y;
q.push(no);
map[x][y]=0;
while(!q.empty())
{
no=q.front();
q.pop();
for(i=0;i<4;i++)
{
ne.x=no.x+move[i][0];
ne.y=no.y+move[i][1];
if(ne.x<0||ne.x>=n||ne.y<0||ne.y>=m)
continue;
if(map[ne.x][ne.y]==1)
{
map[ne.x][ne.y]=0;
q.push(ne);
}
}
}
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int i,j;
scanf("%d%d",&n,&m);
for(i=0;i<n;i++)
for(j=0;j<m;j++)
scanf("%d",&map[i][j]);
sum=0;
for(i=0;i<n;i++)
for(j=0;j<m;j++)
if(map[i][j])
{
sum++;
bfs(i,j);
}
printf("%d\n",sum);
}
}