AcWing 1097. 池塘计数

题目链接 AcWing1097. 池塘计数

经典FloodFill 用bfs做

#include <iostream>
#include <algorithm>
#include <cstring>
#define x first
#define y second

using namespace std;

typedef pair<int, int> PII;

const int N = 1010;

int n, m;
char g[N][N];      //存图
bool st[N][N];   //标记有没有遍历过
PII q[N * N];   //模拟队列

void bfs(int sx, int sy)
{
    st[sx][sy] = true;
    int hh = 0, tt = 0;
    q[0] = {sx, sy};
    
    while(hh <= tt)
    {
        PII t = q[hh ++];
        for(int i = t.x - 1; i <= t.x + 1; i ++)
            for(int j = t.y - 1; j <= t.y + 1; j ++)
            {
                if(i == t.x && j == t.y) continue;
                if(i < 0 || i >= n || j < 0 || j >= m) continue;
                if(g[i][j] == '.' || st[i][j]) continue;
                
                q[++ tt] = {i, j};
                st[i][j] = true;
            }
    }
}

int main()
{
    cin >> n >> m;
    for(int i = 0; i < n; i ++) scanf("%s", g[i]);
    int res = 0;
    for(int i = 0; i < n; i ++)
        for(int j = 0; j < m; j ++)
        {
            if(g[i][j] == 'W' && !st[i][j])
            {
                bfs(i, j);
                res ++;
            }
        }
    cout << res << endl;
    return 0;
}