搜索之连通块(深搜广搜版)
连通块模板用法:
- 输入数据:0/1矩阵,若标准输入不是0/1矩阵,请转换。
转换0/1矩阵,你可能需要形如这样的代码:
char ch
switch(ch) {
case '墙': {s[i][j] = 0;break;}
case '空地': {s[i][j] = 1;break;}
}
- 适合问题:连通块
广度优先搜索版:
#include<bits/stdc++.h>
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 <cstdio>
#include <iostream>
#include <algorithm>
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;
}
Copyright littlefrog(2019)
所有权利保留。