搜索之连通块(深搜广搜版)

连通块模板用法:

  1. 输入数据:0/1矩阵,若标准输入不是0/1矩阵,请转换。
    转换0/1矩阵,你可能需要形如这样的代码:
char ch
switch(ch) {
                case '墙': {s[i][j] = 0;break;}
                case '空地': {s[i][j] = 1;break;} 
            }
  1. 适合问题:连通块

广度优先搜索版:

#include
using namespace std;
struct node
{
    int x;
    int y;
    node(){}
    node(int x1,int y1):x(x1),y(y1){}
 
};
int n,m;
int u[4][2]={{1,0},{-1,0},{0,1},{0,-1}};
 
int head=1;
int tail=1;
int s[105][105];
int sum=0;
void bfs(int x,int y)
{
    queue<node>Q;
    s[x][y]=0;
    Q.push(node(x,y));
    while(!Q.empty()){
        node a=Q.front();
        Q.pop();
        for(int i=0;i<4;i++){
            int xx=u[i][0]+a.x;
            int yy=u[i][1]+a.y;
            if(xx>=0&&xx<n&&yy>=0&&yy<m&&(s[xx][yy])){
                s[xx][yy]=0;
                Q.push(node(xx,yy));
            }
        }
    }
}
int main()
{
    cin>>n>>m;
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            cin>>s[i][j];
        }
    }
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            if(s[i][j]){
                sum++;
                bfs(i,j);
            }
        }
    }
    cout<<sum<<endl;
 
    return 0;
}

深度优先搜索版:

#include 
#include 
#include 
using namespace std;
int n, m;
bool vis[1005][1005];
int maze[1005][1005];
int ans, cnt;
int dir[4][2] = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};
bool in(int x, int y) {
    return x >= 0 && x < n && y >= 0 && y < m;
}
void dfs(int x, int y) {
    cnt++;
    vis[x][y] = true;
    for (int i = 0; i < 4; i++) {
        int tx = x+dir[i][0];
        int ty = y+dir[i][1];
        if(in(tx,ty)&&!vis[tx][ty]&&maze[tx][ty]==1) {
            dfs(tx,ty);
        }
    }
}
int main() {
    cin >> n >> m;
    for (int i = 0;i<n;++i) {
        for(int j = 0;j<m;++j) {
            cin>>maze[i][j];
        }
    }
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            if (maze[i][j] == 1 && !vis[i][j]) {
                cnt = 0;
                dfs(i, j);
                ans ++;
            }
        }
    }
    cout << ans << endl;
    return 0;
}

你可能感兴趣的:(搜索之连通块(深搜广搜版))