##深度优先搜索解决 Lake Counting (POJ No.2386)
问题:
有一个大小为NM的园子,雨后集起了水。八联通的积水被认为是连在一起的。请求出园子里中共有多少水洼?(八联通是指下图中相对W的的部分)
* * *
* w *
* * *
限制条件: N,M<=100;
输入示例:
N=10,M=12;
园子如下图(‘W’表示积水,’.‘表示没有积水)
W . . . . . . . . W W .
. W W W . . . . . W W W
. . . . W W . . . W W .
. . . . . . . . . W W .
. . . . . . . . . W . .
. . W . . . . . . W . .
. W . W . . . . . W W .
**W . W . W . . . . . W . **
**. W . W . . . . . . W . **
. . W . . . . . . . W .
输出示例:3
算法思想:从任意的W开始扫描,不停的把相邻的W替换成“ . ” ,采用DFS算法,一次扫描过后,可以把与这个W相连的W区替换成“ .” 。多次扫描,直到W不再减少时,计算W 个数,即是水洼个数。
代码如下:
#include<iostream>
#define MAX_N 100
#define MAX_M 100
using namespace std;
int N,M;
char garden[MAX_N][MAX_M];
void init(){
cin>>N>>M;
for(int i=0;i<N;i++){
for(int j=0;j<M;j++){
cin>>garden[i][j];
}
}
}
void dfs(int i ,int j){
if(i<0 || i>=N || j<0 || j>=M)
return ;
if( garden[i+1][j] == 'W') { garden[i+1][j]='.'; dfs(i+1,j); } //下
if(garden[i][j+1] == 'W') { garden[i][j+1] = '.' ; dfs(i, j + 1 ); } //右
if(garden[i+1][j+1] == 'W') { garden[i+1][j+1] = '.' ; dfs(i +1, j+1 ); } //右下
if(garden[i+1][j-1] == 'W') { garden[i+1][j-1] = '.' ; dfs(i +1, j-1 ); }//左下
if( garden[i-1][j] == 'W') { garden[i-1][j]='.'; dfs(i-1,j); } //上
if( garden[i][j-1] == 'W') { garden[i][j-1]='.'; dfs(i,j-1); } //左
if( garden[i-1][j-1] == 'W') { garden[i-1][j-1]='.'; dfs(i-1,j-1); } //左上
if( garden[i-1][j+1] == 'W') { garden[i-1][j+1]='.'; dfs(i-1,j+1); } //右上
}
int main(){
int count=0;
init();
for(int i=0;i<N;i++){
for(int j=0;j<M;j++){
if(garden[i][j]=='W'){
dfs( i , j);
count++;
}
}
}
cout<<count<<endl;
return 0;
}