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>
#include<algorithm>
using namespace std;
#define maxn 105

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

void f(int x,int y)
{
    v[x][y]=0;
    int i;
    for(i=0;i<4;i++)
        if(x+dx[i]>=0&&y+dy[i]>=0&&v[x+dx[i]][y+dy[i]]==1)
            f(x+dx[i],y+dy[i]);
}

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