题目描述

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

输入

第一行输入一个整数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题目,dfs题,说白了就是暴力搜索的优化版本,属于不撞南墙不回头的算法,除非有东西阻挡,不然就会一直往下走,直到走完为止。

这题是求水池数目,我们可以用dfs对每个点的上下左右进行搜索,如果某个点的上下左右(某一个或多个)为1,则把这个点设置成0,一趟搜索来,我们可以把跟这个点连接的水池全部标记为0,然后给计数器变量加1,代表这整个为1个水池。只要对所有点都进行一遍dfs,则可以求出所有水池数目。

技巧:
①我们不用二维数组的第一行和第一列,并且将二维数组设置为num105,这样既可以防止数组越界,又可以用第1和101行已经第1列和101列来作为外围边界。
②初始时将二维数组全部设置为0,这样外围就存在边界了。
③将数组定义在外部,这样可以使dfs无序传数组的参数进入。
Coding:

#include <bits/stdc++.h>
#define N 105
using namespace std;
int arr[N][N];

void dfs(int a,int b)
{
    if(arr[a-1][b]==1){arr[a-1][b]=0,dfs(a-1,b);}  //向上搜索
    if(arr[a+1][b]==1){arr[a+1][b]=0,dfs(a+1,b);}  //向下搜索
    if(arr[a][b-1]==1){arr[a][b-1]=0,dfs(a,b-1);}  //向左搜索
    if(arr[a][b+1]==1){arr[a][b+1]=0,dfs(a,b+1);}  //向右搜索
}
int main(void)
{
    int t;
    int n,m;
    cin>>t;
    while(t--)
    {
        int cnt=0;
        cin>>n>>m;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
                cin>>arr[i][j];

        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
            {
                if(arr[i][j]==1)   //代表此点为水池
                {
                    cnt++;
                    dfs(i,j);     //将与该水池相邻的水池都标记为0
                }
            }

            cout<<cnt<<endl;
    }

    return 0;
}

schrodingercatss
8 声望1 粉丝