bfs,dfs整理总结第一弹

这道题本质上就是一个水塘问题

BFS方法

#include<cstdio>
#include<queue>
#include<iostream>
using namespace std;

const int maxn =100;

struct node{
int x,y;
}Node;

int matrix[maxn][maxn];
bool inq[maxn][maxn] ;
int X[4]={0,0,1,-1};
int Y[4]={1,-1,0,0};
int ans=0;
int m,n;

bool judge(int x,int y)
{
    if(x>=n||x<0||y>=m||y<0)  return false;
    if(matrix[x][y]==0||inq[x][y]==true) return false;
    return true;
}

void BFS(int x,int y){
queue<node> Q;
Node.x=x,Node.y=y;
Q.push(Node);
inq[x][y]=true;
while(!Q.empty()){
    node top=Q.front();
    Q.pop();
    for(int i=0;i<4;i++)
    {
        int newX=top.x+X[i];
        int newY=top.y+Y[i];
        if(judge(newX,newY)){
            Node.x=newX,Node.y=newY;
            Q.push(Node);
            inq[newX][newY]=true;
        }

    }
}

}

int main()
{
    cin>>n>>m;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            scanf("%1d",&matrix[i][j]);
        }

    }

    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            if(matrix[i][j]>=1&&matrix[i][j]<=9&&inq[i][j]==false)
            {
                ans++;
                BFS(i,j);
            }
        }
    }
   cout<<ans<<endl;
   return 0;

}

DFS方法

#include<bits/stdc++.h>
using namespace std;
int n,m;
int matrix[105][105];
bool inq[105][105];
int X[4]={0,0,1,-1};
int Y[4]={1,-1,0,0};


void read()
{
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            scanf("%1d",&matrix[i][j]);
        }
    }
}

int dfs(int x,int y)
{
    inq[x][y]=true;

        for(int i=0;i<4;i++)
        {
            x=x+X[i];
            y=y+Y[i];
           if(matrix[x][y]>=1&&matrix[x][y]<=9&&inq[x][y]==false)
           {
               dfs(x,y);
           }x-=X[i],y-=Y[i];
        }

}

int main()
{
    cin>>n>>m;
    int ans=0;
    read();
    for(int i=0;i<n;i++ )
    {
        for(int j=0;j<m;j++)
        {
            if(matrix[i][j]>=1&&matrix[i][j]<=9&&inq[i][j]==false)
           {
             ans++;
             dfs(i,j);
           }
        }
    }
    cout<<ans<<endl;
}