ACM 水池数目

水池数目

时间限制：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 <vector>
#include <algorithm>

using namespace std;

typedef vector<vector<int> > Pool;

const int dx[] = {0,1,0,-1};
const int dy[] = {1,0,-1,0};

void dfs(Pool& pool,int x,int y){
    pool[x][y]=0;
    for(int i = 0 ; i < 4 ; ++ i){
        int newx = x+dx[i],newy = y+dy[i];
        if(pool[newx][newy] == 1){
            dfs(pool,newx,newy);
        }
    }
}
int main(){
    int N;
    cin >> N;
    while(N--){
        int m,n;
        cin >> m >> n;
        Pool pool(m+2,vector<int>(n+2,-1));
        for(int i = 1; i <= m; ++ i){
            for(int j = 1; j <= n; ++ j){
                cin >>  pool[i][j];
            }
        }
        int res = 0;
        for(int i = 1; i <= m ; ++ i){
            for(int j = 1 ; j <= n ; ++ j){
                if(pool[i][j]==1){
                    res++;
                    dfs(pool,i,j);
                }
            }
        }
        cout<<res<<endl;
    }
}