暑期算法心得-----BFS(一)

输入:一个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;

}

猜你喜欢

转载自blog.csdn.net/chengsilin666/article/details/81155857
今日推荐