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

农夫约翰有一片 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

1097. 池塘计数 C++ FloodFill算法_第1张图片
FloodFill思想:每次采用bfs对’W’进行广度优先搜索bfs,并把搜索到的’W’变成’.’,最后答案就是bfs的次数。

#include
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;
}

你可能感兴趣的:(算法,bfs)