NYOJ-水池数目（bfs）

水池数目

题目描述:

南阳理工学院校园里有一些小河和一些湖泊，现在，我们把它们通一看成水池，假设有一张我们学校的某处的地图，这个地图上仅标识了此处是否是水池，现在，你的任务来了，请用计算机算出该地图中共有几个水池。

输入描述:

第一行输入一个整数N，表示共有N组测试数据
每一组数据都是先输入该地图的行数m(0<m<100)与列数n(0<n<100)，然后，输入接下来的m行每行输入n个数，表示此处有水还是没水（1表示此处是水池，0表示此处是地面）

输出描述:

输出该地图中水池的个数。
要注意，每个水池的旁边（上下左右四个位置）如果还是水池的话的话，它们可以看做是同一个水池。

样例输入:

2
3 4
1 0 0 0 
0 0 1 1
1 1 1 0
5 5
1 1 1 1 0
0 0 1 0 1
0 0 0 0 0
1 1 1 0 0
0 0 1 1 1

样例输出:

2
3

解题思路：

这是个bfs模板，就是看有多少片1

程序代码：

#include<stdio.h>
#include<queue>
using namespace std;
struct node{
    int x;
    int y;
};
int m,n;
int a[110][110];
int bfs(int x,int y);
int main()
{
    int T,i,j,sum;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&m,&n);
        for(i=0;i<m;i++)
            for(j=0;j<n;j++)
                scanf("%d",&a[i][j]);
        sum=0;
        for(i=0;i<m;i++)
            for(j=0;j<n;j++)
                if(a[i][j]==1)
                {
                    sum++;
                    bfs(i,j);
                }
        printf("%d\n",sum);
    }
    return 0;
}
int bfs(int x,int y)
{
    int i;
    int next[4][2]={0,1,1,0,0,-1,-1,0};
    queue<node>q;
    node A,B;
    A.x=x;
    A.y=y;
    q.push(A);
    while(!q.empty())
    {
        A=q.front();
        q.pop();
        for(i=0;i<4;i++)
        {
            B.x=A.x+next[i][0];
            B.y=A.y+next[i][1];
            if(B.x<0||B.y<0||B.x>m-1||B.y>n-1)
                continue;
            if(a[B.x][B.y]==1)
            {
                a[B.x][B.y]=0;
                q.push(B);
            }
        }
    }
}