ACM 水池數目

OpenSoucre發表於2014-04-30

水池數目

時間限制: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;
    }
}