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

 
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

int visit[105][105];
int s[105][105], m, n;
const int dx[4] = {-1,0,1,0};
const int dy[4] = {0,1,0,-1};

void dfs(int i,int j)
{
    visit[i][j] = 1;
    for(int k = 0; k < 4; k++)
    {
        int nx = i + dx[k];
        int ny = j + dy[k];
        if((nx >= 0) && (nx < m) && (ny >= 0) && (ny < n) && (visit[nx][ny] == 0) && (s[nx][ny] == 1))
        {
            dfs(nx,ny);    
        }        
    }
}


int main(){
    int cou, test;
    cin >> test;
    
    while(test--)
    {
        cou = 0;
        memset(visit, 0, sizeof(visit));
        memset(s, 0, sizeof(s)); 
        
        cin >> m >> n;                                                            
        for(int i = 0; i < m; i++)
            for(int j = 0; j < n; j++)
                cin >> s[i][j];
        
        for(int i = 0; i < m; i++)
            for(int j = 0; j < n; j++)
                if(visit[i][j] == 0 && s[i][j] == 1)
                {
                    cou++;
                    dfs(i, j);   
                }
                        
        printf("%d\n", cou);
    }
    return 0;
}