简单BFS 模板题 PTA 岛屿数量

简单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;
}