NYOJ-标题27水池数目

NYOJ---题目27水池数目

水池数目

描述

南阳理工学院校园里有一些小河和一些湖泊，现在，我们把它们通一看成水池，假设有一张我们学校的某处的地图，这个地图上仅标识了此处是否是水池，现在，你的任务来了，请用计算机算出该地图中共有几个水池。

输入

第一行输入一个整数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题，从一个点出发，把所有是1的点都变成0，经历几次搜索，就是有几个水池；

代码如下：

#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
int dx[]={1,0,-1,0};
int dy[]={0,1,0,-1};
int vis[110][110];
int n,m;
inline bool in(int x,int y)
{
    if(x>=0&&x<m&&y>=0&&y<n)
        return true;
    return false;
}
void dfs(int x,int y)
{
    vis[x][y]=0;
    int newx,newy;
    for(int i=0;i<4;i++)
    {
        newx=x+dx[i];
        newy=y+dy[i];
        if(vis[newx][newy]==1&&in(newx,newy))
        {
            dfs(newx,newy);
        }
    }
}
int main()
{
    int test;
    cin>>test;
    while(test--)
    {

        cin>>m>>n;
        int count=0;
        for(int i=0;i<m;i++)
            for(int j=0;j<n;j++)
        {
            cin>>vis[i][j];
        }
        for(int i=0;i<m;i++)
            for(int j=0;j<n;j++)
        {
            if(vis[i][j]==1)
            {
                dfs(i,j);
                count++;
            }
        }
        printf("%d\n",count);

    }
}