输入:一个0-1矩阵
输出:该矩阵所包含的块的个数。
下面是BFS的一个典型模板。
#include <iostream>
#include <queue>
#include <cstdio>
using namespace std;
const int maxn=100;
int X[4]={0,0,1,-1};
int Y[4]={1,-1,0,0};
int G[maxn][maxn];
int visited[maxn][maxn];
int ans=0;
int m,n;
struct Point
{
int x;
int y;
};
Point p;
int isOK(int x,int y)
{
if(x<0||y<0||x>=m||y>=n)
{
return 0;
}
if(G[x][y]==0||visited[x][y]==1)
{
return 0;
}
return 1;
}
void BFS(int x,int y)
{
Point temp;
queue<Point> q;
int i,j;
p.x=x;
p.y=y;
q.push(p);
visited[x][y]=1; //最先进入队列的元素需要添加相应的访问标记
while(!q.empty())
{
Point top=q.front();
q.pop();
for(i=0;i<4;i++)
{
int newX=top.x+X[i]; //增量数组对应的top不同于下面判断语句中的temp,在写代码的时候要注意将两者区分开来
int newY=top.y+Y[i];
if(isOK(newX,newY)) //进行BFS的先决条件
{
temp.x=newX;
temp.y=newY;
q.push(temp);
visited[temp.x][temp.y]=1;
}
}
}
}
int main()
{
int i,j,k;
scanf("%d %d",&m,&n);
for(i=0;i<m;i++)
for(j=0;j<n;j++)
{
scanf("%d",&G[i][j]);
}
for(i=0;i<m;i++)
for(j=0;j<n;j++)
{
visited[i][j]=0;
}
for(i=0;i<m;i++)
for(j=0;j<n;j++)
{
if(visited[i][j]!=1&&G[i][j]==1)
{
ans++;
BFS(i,j);
}
}
printf("%d\n",ans);
return 0;
}