首页 > 代码库 > NYOJ 27 水池数目

NYOJ 27 水池数目

水池数目

时间限制:3000 ms  |  内存限制:65535 KB
难度:4
描述
南阳理工学院校园里有一些小河和一些湖泊,现在,我们把它们通一看成水池,假设有一张我们学校的某处的地图,这个地图上仅标识了此处是否是水池,现在,你的任务来了,请用计算机算出该地图中共有几个水池。
输入
第一行输入一个整数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
思路:简单的深搜
AC代码:
#include<iostream>
#include<string.h>
using namespace std;
 int dir[4][2]={{-1,0},{1,0},{0,-1},{0,1}};
 int a[100][100],vis[100][100],m,n;
 void dfs(int i,int j)
 {
     a[i][j]=0;
     int k,x,y;
    for(int k=0;k<4;k++)
    {
        x= i + dir[k][0];  
        y = j + dir[k][1];  
        if (x>=0 &&x<m && y>=0 && y<n && a[x][y] && !vis[x][y])  
            {
                       dfs(x,y);  
                       vis[x][y]=1;
            }     
    }
 }
int main()
{
   int N,i,j,sum;
   cin>>N;
   while(N--)
   {
             sum=0;
             memset(a,0,sizeof(a));
             memset(vis,0,sizeof(vis));
        cin>>m>>n;
         for(i=0;i<m;i++)
          for(j=0;j<n;j++)
            cin>>a[i][j]; 
            for(i=0;i<m;i++)
              for(j=0;j<n;j++)
              {
                 if(a[i][j]==1&&vis[i][j]==0)
                 {
                   sum++;
                   dfs(i,j);
                   vis[i][j]=1;
                 }
              }
              cout<<sum<<endl;
              
                
   }
  // system("pause");
   return 0;
}