1097. 池塘计数 C++ FloodFill算法

农夫约翰的池塘计数

最新推荐文章于 2025-10-19 18:14:55 发布

原创最新推荐文章于 2025-10-19 18:14:55 发布 · 1.4k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #bfs

本博客介绍了一个基于广度优先搜索(BFS)的算法，用于计算一个由字符矩阵表示的矩形土地上形成的池塘数量。池塘是由相连的积水单元格构成的，每个单元格可以与其八个邻近单元格相连。通过将'W'(表示积水)转换为'.'(表示干燥)，算法能够准确地计算出矩阵中独立池塘的数量。

农夫约翰有一片 N∗M 的矩形土地。
最近，由于降雨的原因，部分土地被水淹没了。
现在用一个字符矩阵来表示他的土地。
每个单元格内，如果包含雨水，则用”W”表示，如果不含雨水，则用”.”表示。
现在，约翰想知道他的土地中形成了多少片池塘。
每组相连的积水单元格集合可以看作是一片池塘。
每个单元格视为与其上、下、左、右、左上、右上、左下、右下八个邻近单元格相连。
请你输出共有多少片池塘，即矩阵中共有多少片相连的”W”块。

输入样例：

10 12
W........WW.
.WWW.....WWW
....WW...WW.
.........WW.
.........W..
..W......W..
.W.W.....WW.
W.W.W.....W.
.W.W......W.
..W.......W.

输出样例：

3

FloodFill思想:每次采用bfs对’W’进行广度优先搜索bfs，并把搜索到的’W’变成’.’，最后答案就是bfs的次数。

#include<bits/stdc++.h>
using namespace std;

const int N = 1010;
typedef pair<int,int> PII;
int n,m;
char farm[N][N];
int dx[8]={1,1,1,0,0,-1,-1,-1};
int dy[8]={1,0,-1,1,-1,1,0,-1};

void bfs(int x,int y)
{
    queue<PII> q;
    q.push({x,y});
    while(q.size())
    {
        auto t = q.front();
        q.pop();
        for(int i=0;i<8;i++)
        {
            int xi = t.first + dx[i],yi = t.second + dy[i];
            if(xi<0 || yi<0 || xi>=n || yi>=m) continue;
            if(farm[xi][yi]=='.') continue;
            
            farm[xi][yi]='.';
            q.push({xi,yi});
        }
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++)
    {
        scanf("%s",farm[i]);
    }
    int ans=0;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
           if(farm[i][j]=='W')
           {
               bfs(i,j);
               ans++;
           }
        }
    }
    cout<<ans;
    return 0;
}