NYOJ27.水池数目-DFS求连通块

Posted 2020-11-10 Persistent.

水池数目

时间限制：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

代码:

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<cmath>
 5 #include<algorithm>
 6 #include<queue>
 7 #include<vector>
 8 using namespace std;
 9 typedef long long ll;
10 const int maxn=100+10;
11 int a[maxn][maxn];
12 int dir[4][2]={-1,0,1,0,0,1,0,-1};
13 int n,m,num;
14 void DFS(int x,int y){
15     a[x][y]=0;
16     for(int i=0;i<4;i++){
17         int dx=x+dir[i][0];
18         int dy=y+dir[i][1];
19         if(dx>=0&&dy>=0&&dx<n&&dy<m&&a[dx][dy])
20             DFS(dx,dy);
21     }
22 }
23 int main(){
24     int t;
25     scanf("%d",&t);
26     while(t--){
27        num=0;
28        scanf("%d%d",&n,&m);
29        memset(a,0,sizeof(a));
30        for(int i=0;i<n;i++){
31         for(int j=0;j<m;j++)
32             scanf("%d",&a[i][j]);
33        }
34        for(int i=0;i<n;i++){
35         for(int j=0;j<m;j++){
36             if(a[i][j]==1){
37                 DFS(i,j);
38                 num++;
39             }
40         }
41        }
42        printf("%d\n",num);
43     }
44     return 0;
45 }