【ACM】算法题-水池数目(C++)(宽度搜索)(队列)(递归)

水池数目

题目

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

输入

第一行输入一个整数N，表示共有N组测试数据
每一组数据都是先输入该地图的行数n(0<n<100)与列数m(0<m<100)，然后，输入接下来的m行每行输入n个数，表示此处有水还是没水（1表示此处是水池，0表示此处是地面）

输出

输出该地图中水池的个数。
要注意，每个水池的旁边（上下左右四个位置）如果还是水池的话的话，它们可以看做是同一个水池。

思路小导

本题不管是深度搜索还是宽度搜索都能够完成，重点是在于搜索套路与细节处理，此处我应用的是宽度搜索

#include<bits/stdc++.h>
#include<iostream>
using namespace std;
#define endl '\n'
using ll=long long;
const size_t maxn = 101;
int a[maxn][maxn], book[maxn][maxn];//定义要检索的图（二维数组a）,和标记数组book保证每个元素不被重复访问
int fx[4][2] = {{0,  1},
                {0,  -1},
                {1,  0},
                {-1, 0}};//定义方向数组,可移动的方位上下左右
int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    int area = 0;//定义区域数目
    int T;
    cin >> T;
    while (T--) {
        int m, n;
        cin >> m >> n;
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                cin >> a[i][j];
                book[i][j] = 0;//输入二维数组后标记数组也要设好初值
            }
        }
        area = 0;//每次循环要更新区域数目
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (a[i][j] == 0 || book[i][j] == 1)continue;//先过滤掉区域内是水的范围和已经被标记过的情况
                ++area;//区域数目增加
                queue<pair<int, int>> Q;//定义pair队列，利用队列的进出来实现
                Q.push(make_pair(i, j));
                book[i][j] = 1;//压入队列之后就得标记数组
                while (Q.size()) {//在队列不为空的情况也可以写成!Q.empty()
                    pair<int, int> pt = Q.front();//
                    int x = pt.first;//
                    int y = pt.second;//定义横纵坐标
                    for (int k = 0; k < 4; k++) {//四个方向的搜索
                        int nx = x + fx[k][0];
                        int ny = y + fx[k][1];
                        if (nx < 0 || nx >= m || ny < 0 || ny >= n || book[nx][ny] || a[nx][ny] == 0)continue;
                        Q.push(make_pair(nx, ny));//过滤掉不合法的情况后，把坐标压入队列
                        book[nx][ny] = 1;//标记数组
                    }
                    Q.pop();//移出队列
                }
            }
        }
        cout << area << endl;//输出区域数目
    }
    return 0;
}

小结

牢记宽度搜索的套路

认真把握过程的细节

不断检测自己的代码