简单BFS 模板题 PTA 岛屿数量
给你一个由 1(陆地)和 0(水)组成的n*m的二维网格,请你计算网格中岛屿的数量num。
岛屿总是被水包围,并且每座岛屿只能由水平方向或竖直方向上相邻的陆地连接形成。例如
例 | 例 | 例 |
---|---|---|
0 | 0 | 0 |
0 | 1 | 0 |
0 | 0 | 1 |
此为两个岛屿
此外,你可以假设该网格的四条边均被水包围。1<=n,m<=100
输入格式:
第一行中给出网格长宽n,m
接下来的n行表示网格情况
输出格式:
岛屿个数num
输入样例:
在这里给出一组输入。例如:
4 5
1 1 0 0 0
1 1 0 0 0
0 0 1 0 0
0 0 0 1 1
输出样例:
在这里给出相应的输出。例如:
3
#include<bits/stdc++.h>
using namespace std;
struct node{
int x,y;
};
queue<node>q;
int a[110][110];
int b[110][110];
int d[4][2]={{1,0},{0,1},{-1,0},{0,-1}};
int n,m,cnt=0;
void bfs(int x,int y){
b[x][y]=cnt;
node n1;
n1.x=x;n1.y=y;
q.push(n1);
while(!q.empty()){
node n2=q.front();
int x1=n2.x;int y1=n2.y;
for(int i=0;i<4;i++){
int xx=x1+d[i][0];int yy=y1+d[i][1];
if(a[xx][yy]==1&&b[xx][yy]==0&&xx>0&&xx<=n&&yy>0&&yy<=m){
b[xx][yy]=cnt;
node n3;
n3.x=xx;n3.y=yy;
q.push(n3);
}
}
q.pop();
}
}
int main(){
cin>>n>>m;
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
cin>>a[i][j];
}
}
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
if(a[i][j]==1&&b[i][j]==0){
cnt++;
bfs(i,j);
}
}
}
cout<<cnt<<endl;
return 0;
}