1097. 池塘计数Flood Fill 搜索

Posted 2021-11-04 幽殇默

https://www.acwing.com/problem/content/1099/
很裸的一个求连通块的问题
用bfs()写 缺点: 对于本题代码有点长

#pragma GCC optimize(2)
#include<bits/stdc++.h>
using namespace std;
const int N=1e3+10;
char a[N][N];
int st[N][N],n,m,ans;
int dx[8]={-1,-1,-1,0,0,1,1,1};
int dy[8]={-1,0,1,-1,1,-1,0,1};
void flood(int x,int y)
{
    queue< pair<int,int> >q; q.push({x,y});
    while(q.size())
    {
        x=q.front().first;
        y=q.front().second;  q.pop();
        if(st[x][y]) continue;
        st[x][y]=1;
        for(int i=0;i<8;i++)
        {
            int tempx=x+dx[i];
            int tempy=y+dy[i];
            if(tempx>=0&&tempx<n&&tempy>=0&&tempy<m&&!st[tempx][tempy]&&a[tempx][tempy]=='W')
            q.push({tempx,tempy});
        }
    }
}
int main(void)
{
    cin>>n>>m;
    for(int i=0;i<n;i++)
        for(int j=0;j<m;j++)
            cin>>a[i][j];
    for(int i=0;i<n;i++)
    {    
       for(int j=0;j<m;j++) 
       if(st[i][j]==0&&a[i][j]=='W') flood(i,j),ans++;
    }
    cout<<ans<<endl;
    return 0;
}

DFS来写代码短。

#include<bits/stdc++.h>
using namespace std;
const int N=1e3+10;
string s[N];
int n,m,cnt;
int dx[8]={-1,-1,-1,0,0,1,1,1};
int dy[8]={-1,0,1,-1,1,-1,0,1};
void dfs(int x,int y)
{
    s[x][y]='.';
    for(int i=0;i<8;i++)
    {
        int tempx=x+dx[i];
        int tempy=y+dy[i];
        if(tempx<0||tempx>=n||tempy<0||tempy>=m||s[tempx][tempy]=='.') continue;
        dfs(tempx,tempy);
    }
}
int main(void)
{
    cin>>n>>m;
    for(int i=0;i<n;i++) cin>>s[i];
    for(int i=0;i<n;i++)
        for(int j=0;j<m;j++) if(s[i][j]=='W') dfs(i,j),cnt++;
    cout<<cnt<<endl;
    return 0;
}