nyoj 27 水池数目（dfs求连通块）

「爱情、让人受尽委屈。」 2024-02-17 20:56 21阅读 0赞

### 水池数目 ###

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

分析：求连通块的个数，简单dfs

AC代码：

#include<cstdio>
    #include<cstring>
    using namespace std;
    const int maxn=100+2;
    int g[maxn][maxn];
    int vis[maxn][maxn];
    int n,m;
    int dir[4][2]={
       {-1,0},{1,0},{0,-1},{0,1}};
    void dfs(int i,int j){
    	
    	vis[i][j]=1;
    	
    	for(int k=0;k<4;k++)
          {
          	int x=i+dir[k][0];
          	int y=j+dir[k][1];
          	if(x<0 || x>=n || y<0 || y>=m)continue;
          	
          	if(!vis[x][y] && g[x][y]==1){
          		dfs(x,y);
    		  }
    	  }
    }
    int main(){
    	int T;
    	scanf("%d",&T);
    	while(T--){
    		scanf("%d%d",&n,&m);
    		for(int i=0;i<n;i++){
    			for(int j=0;j<m;j++)
    			scanf("%d",&g[i][j]);
    		}
    		
    		int tol=0;
    		memset(vis,0,sizeof(vis));
    		for(int i=0;i<n;i++){
    			for(int j=0;j<m;j++){
    				if(!vis[i][j] && g[i][j]==1){
    					dfs(i,j);
    					tol++;
    				}
    			}
    		}
    		
    		printf("%d\n",tol);
    	}
    	return 0;
    }