南陽OJ 題目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

簡單DFS水過。

#include<iostream>
#include<cstring>
using namespace std;
int n,m;
int vis[101][101],map[101][101];
void dfs(int i,int j)
{
	if(!vis[i+1][j]&&map[i+1][j])
	vis[i+1][j]=1,dfs(i+1,j);
	if(!vis[i-1][j]&&map[i-1][j])
	vis[i-1][j]=1,dfs(i-1,j);
	if(!vis[i][j-1]&&map[i][j-1])
	vis[i][j-1]=1,dfs(i,j-1);
	if(!vis[i][j+1]&&map[i][j+1])
	vis[i][j+1]=1,dfs(i,j+1);
}
int main()
{
	int T;
	cin>>T;
	while(T--){
	int ans=0;
	cin>>n>>m;
	memset(vis,0,sizeof(vis));
	memset(map,0,sizeof(map));
	for(int i=1;i<=n;i++)
	for(int j=1;j<=m;j++)
	cin>>map[i][j];
    for(int i=1;i<=n;i++)
	for(int j=1;j<=m;j++)
	if(map[i][j]&&vis[i][j]==0)
	vis[i][j]=1,ans++,dfs(i,j);
	cout<<ans<<endl;
}
return 0;
}