矩阵中块的个数计算--BFS

矩阵中块的个数计算–BFS

问题描述:

给出一个m*n的矩阵，矩阵中的元素为0或1.称位置（x,y）与其上下左右四个位置(x,y+1),(x,y-1),(x+1,y),(x-1,y)是相邻的。如果矩阵中有若干个1是相邻的(不必两两相邻)，那么称这些1构成了一个“块”。求给定矩阵中"块"的个数

#include<bits/stdc++.h>
using namespace std;
const int maxn=100;

struct Node{
	int x,y; //位置(x,y); 
}node; 

int n,m;   //矩阵大小为n*m;
int matrix[maxn][maxn];  //01矩阵 
bool inq[maxn][maxn]={false}; //记录位置(x,y)是否已入过队
int X[4]={0,0,1,-1};
int Y[4]={1,-1,0,0};

bool judge(int x,int y){
	//越界返回false;
	if(x>=n||x<0||y>=m||y<0)return false;
	//当前位置为0，或(x,y)已入过队，返回false
	if(matrix[x][y]==0||inq[x][y]==true)return false;
	//以上都满足,返回true
	return true;
} 

//BFS函数访问位置(x,y)所在的块，将该块中所有的"1"的inq都设置为true
void BFS(int x,int y){
	queue<Node>Q; 		//定义队列 
	node.x=x;			//当前结点的位置为(x,y) 
	node.y=y;			
	inq[x][y]=true;		//设置(x,y)已入过队 
	Q.push(node);		//将结点入队 
	while(!Q.empty()){
		Node top=Q.front();
		Q.pop();
		for(int i=0;i<4;i++){		//循环4次,得到4个相邻的位置 
			int newX=top.x+X[i];
			int newY=top.y+Y[i];
			if(judge(newX,newY)){
				//设置Node的坐标为(newX,newY);
				 node.x=newX;
				 node.y=newY;
				 Q.push(node);		//将节点Node入队 
				 inq[newX][newY]=true; //设置位置(newX,newY)已入队 
			}
		} 
	}
} 

int main(){
	scanf("%d%d",&n,&m);
	for(int i=0;i<n;i++)
		for(int j=0;j<m;j++){
			scanf("%d",&matrix[i][j]);
		}
	int ans=0;	 //存放块的个数 
	for(int x=0;x<n;x++){
		for(int y=0;y<m;y++){
			if(matrix[x][y]==1&&inq[x][y]==false){
				ans++;
				BFS(x,y);
			}
		}
	} 
	printf("%d\n",ans);
	return 0;
}