NYOJ--水池数目

//NYOJ--水池数目
#include<iostream>
#include<cstring>
const int MAX_SIZE{102};
using namespace std;
void dfs(int,int);
int map[MAX_SIZE][MAX_SIZE];
int main(){
    //freopen("input.txt","r",stdin);
    int N;
    cin>>N;
    while(N--) {
        int count=0;
        int row,col;
        cin>>row>>col;
        memset(map,0,sizeof(map));
        for(int i=1;i<row+1;++i)
            for(int j=1;j<col+1;++j){
                cin>>map[i][j];
            }
        for(int i=1;i<row+1;++i)
            for(int j=1;j<col+1;++j){
                if(map[i][j]){
                    count++;
                    map[i][j]=0;
                    dfs(i,j);
                }
            }
        cout<<count<<endl;
    }
    return 0;
}
void dfs(int i,int j){
    if(map[i][j-1]) { map[i][j-1]=0; dfs(i,j-1); }
    if(map[i][j+1]) { map[i][j+1]=0; dfs(i,j+1); }
    if(map[i-1][j]) { map[i-1][j]=0; dfs(i-1,j); }
    if(map[i+1][j]) { map[i+1][j]=0; dfs(i+1,j); }
}
原文地址:https://www.cnblogs.com/langyao/p/7251910.html