DFS洛谷4961(求联通块)

说实话这个题审题把我卡了半天,还是我太菜

直接上代码吧

偷个懒用万能库。 

#include"bits/stdc++.h"
using namespace std;
int mp[1010][1010];
int n,m;
int dx[8]={1,1,1,0,0,-1,-1,-1},dy[8]={-1,0,1,-1,1,-1,0,1};
int ans;
void dfs(int x,int y)
{   
    mp[x][y]=3;
    for(int i=0;i<8;i++)
    {
        if(x+dx[i]>=n||x+dx[i]<0||y+dy[i]>=m||y+dy[i]<0||mp[x+dx[i]][y+dy[i]]==3||mp[x+dx[i]][y+dy[i]]==2)   continue;
        dfs(x+dx[i],y+dy[i]);
    }
}
int main()
{
    cin>>n>>m;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)    scanf("%d",&mp[i][j]);
    }
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            if(!mp[i][j])
            {
                for(int k=0;k<8;k++)
                {
                    if(i+dx[k]<0||i+dx[k]>=n||j+dy[k]<0||j+dy[k]>=m)  continue;
                    if(mp[i+dx[k]][j+dy[k]]==1) mp[i][j]=2;
                }
            }
        }
    }
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            if(mp[i][j]==2)
            {   
                int flag=0;
                for(int k=0;k<8;k++)
                {
                    if(mp[i+dx[k]][j+dy[k]]!=0||i+dx[k]<0||i+dx[k]>=n||j+dy[k]<0||j+dy[k]>=m)  continue;
                    flag=1;
                }
                if(flag==0) ans++;
            }
        }
    }
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            if(mp[i][j]==0)
            {
                dfs(i,j);
                ans++;
            }
        }
    }
    cout<<ans<<endl;
}

posted @ 2022-04-18 23:06  cxy8  阅读(21)  评论(0)    收藏  举报