ACM--DFS--nyoj 27--水池数目

nyoj地址：http://acm.nyist.edu.cn/JudgeOnline/problem.php?pid=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

#include<iostream>
#include<stdio.h>
using namespace std;
int MAP[110][110];
int move[4][2]={-1, 0, 0, 1, 1, 0, 0, -1};
/**
  用来初始化地图
*/
void initMap(int rowNum,int colNum){
    int i;
    for(i=0;i<rowNum+2;i++){
        MAP[i][0]=MAP[i][colNum+1]=0;
    }
    for(i=0;i<colNum+2;i++){
        MAP[0][i]=MAP[rowNum+1][i]=0;
    }
}
/**
  深度搜索
*/
void DFS(int n,int m){
    MAP[n][m]=0;
    int i;
    for(i=0;i<4;i++){
        if(MAP[n+move[i][0]][m+move[i][1]]==1){
            DFS(n+move[i][0],m+move[i][1]);
        }
    }
}
int main(){
   int n,rowNum,colNum,i,j,result;
   cin>>n;
   while(n--){
        //录入行列
      cin>>rowNum>>colNum;
      initMap(rowNum,colNum);
      //将地图录入
      for(i=1;i<=rowNum;i++){
        for(j=1;j<=colNum;j++){
            cin>>MAP[i][j];
        }
      }
      result=0;
     //每一个元素进行深度搜索
     for(i=1;i<=rowNum;i++){
        for(j=1;j<=colNum;j++){
           if(MAP[i][j]==1){
              DFS(i,j);
              result++;
           }
        }
      }
      printf("%d\n",result);
   }
   return 0;
}