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<stdio.h>
#include<string.h>
int map[105][105];
int m,n;
bool visit[105][105];
int dfs(int a,int b){
	if(0<=a&&a<m&&0<=b&&b<n){//刚开始的时候这里的条件设置为了0<=a<m&&0<=b<n,运行一直有一组错误数据比如地图为1 0或0 1的时候
		if(map[a][b]&&!visit[a][b]){//是水池并且没搜索过
		visit[a][b]=1;
	    dfs(a-1,b);//上 	
       	dfs(a+1,b);//下 
        dfs(a,b-1);//左 
	    dfs(a,b+1);//右 
	    //dfs(a-1,b-1);//左上 
        //dfs(a+1,b-1);//左下 
	    //dfs(a+1,b+1); //右下 
	    //dfs(a-1,b+1);//右上 
	    return 1;
       }
	}
	return 0;
}
int main(){
	int N;
	scanf("%d",&N);
	while(N--){
		scanf("%d%d",&m,&n);
		for(int i=0;i<m;i++){
			for(int j=0;j<n;j++)
			  scanf("%d",&map[i][j]);
		}
		memset(visit,0,sizeof(visit));
		int count=0;
		for(int i=0;i<m;i++){
			for(int j=0;j<n;j++){
			     int a=dfs(i,j);count+=a;
			    //printf("**%d\n",a);
			}
			
		}
		printf("%d\n",count);
	}
} 

//上面的优化一下：

#include<stdio.h>
#include<string.h>
int map[105][105];
int m,n;
//bool visit[105][105];
int dfs(int a,int b){
	if(0<=a&&a<m&&0<=b&&b<n){
		if(map[a][b]){//是水池并且没搜索过
		map[a][b]=0;
		//visit[a][b]=1;&&!visit[a][b]
	    
		dfs(a-1,b);//上 	
       	dfs(a+1,b);//下 
        dfs(a,b-1);//左 
	    dfs(a,b+1);//右
	    //dfs(a-1,b-1);//左上 
        //dfs(a+1,b-1);//左下 
	    //dfs(a+1,b+1); //右下 
	    //dfs(a-1,b+1);//右上 
	   // return 1;
       }
	}
	//return 0;
}
int main(){
	int N;
	scanf("%d",&N);
	while(N--){
		scanf("%d%d",&m,&n);
		for(int i=0;i<m;i++){
			for(int j=0;j<n;j++)
			  scanf("%d",&map[i][j]);
		}
		memset(visit,0,sizeof(visit));
		int count=0;
		for(int i=0;i<m;i++){
			for(int j=0;j<n;j++){
			     if(map[i][j]){
			     	count++;
			     	dfs(i,j);
				 } 
			    //printf("**%d\n",a);
			}
			
		}
		printf("%d\n",count);
	}
}