深度优先搜索解决Lake Counting (POJ No.2386)

深度优先搜索解决 Lake Counting (POJ No.2386)

问题:
有一个大小为N*M的园子,雨后集起了水。八联通的积水被认为是连在一起的。请求出园子里中共有多少水洼?(八联通是指下图中相对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;
}

猜你喜欢

转载自blog.csdn.net/qq_28120673/article/details/81015939
今日推荐