POJ 2386 Lake Counting

题目

源地址:

http://poj.org/problem?id=2386

理解

当年学习DFS的AC掉的一道水题。DFS的经典模板,通过找出全部的相连池塘块来确定有几块池塘。

代码

原始版本

#include  <cstdio>
#include  <cstring>
using namespace std;

int x, y, i, j, ans;
char Map[100][100];
int visited[100][100];

int dfs(int i, int j)
{
    if (!visited[i][j] && Map[i][j] == 'W')
    {
        visited[i][j] = 1;
        if (j + 1 < y && Map[i][j + 1] == 'W')     dfs(i, j + 1);
        if (j + 1 < y && i - 1 >= 0 && Map[i - 1][j + 1] == 'W')    dfs(i - 1, j + 1);
        if (i - 1 >= 0 && Map[i - 1][j] == 'W')    dfs(i - 1, j);
        if (i - 1 >= 0 && j - 1 >= 0 && Map[i - 1][j - 1] == 'W')   dfs(i - 1, j - 1);
        if (j - 1 >= 0 && Map[i][j - 1] == 'W')     dfs(i, j - 1);
        if (j - 1 >= 0 && i + 1 < x && Map[i + 1][j - 1] == 'W')     dfs(i + 1, j - 1);
        if (i + 1 < x && Map[i + 1][j] == 'W')      dfs(i + 1, j);
        if (i + 1 < x && j + 1 < y && Map[i + 1][j + 1] == 'W')     dfs(i + 1, j + 1);
    }
    return 0;
}

int main()
{
    scanf("%d%d", &x, &y);

    ans = 0;
    memset(visited, 0, sizeof(visited));
    for (i = 0; i < x; i++)    scanf("%s", Map[i]);
    for (i = 0; i < x; i++)
    {
        for (j = 0; j < y; j++)
        {
            if (Map[i][j] == 'W' && !visited[i][j])
            {
                dfs(i, j);
                ans ++ ;
            }
        }
    }
    printf("%d\n", ans);

    return 0;
}

优化版本


#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <ctime>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <deque>
#include <list>
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <numeric>
#include <iomanip>
#include <bitset>
#include <sstream>
#include <fstream>
#define debug puts("-----")
#define pi (acos(-1.0))
#define eps (1e-8)
#define inf (1<<30)
#define MAXN 110
using namespace std;

int n, m;
char field[MAXN][MAXN + 1];

void dfs(int x, int y)
{
    field[x][y] = '.';
    for (int dx = -1; dx <= 1; dx++)
    {
        for (int dy = -1; dy <= 1; dy++)
        {
            int nx = x + dx, ny = y + dy;
            if (0 <= nx && nx < n && 0 <= ny && ny < m && field[nx][ny] == 'W')
                dfs(nx, ny);
        }
    }
    return;
}

int main(int argc, char const *argv[])
{

    int res = 0;
    cin >> n >> m;
    for (int i = 0; i < n; i++)
        cin >> field[i];
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < m; j++)
        {
            if (field[i][j] == 'W')
            {
                dfs(i, j);
                res++;
            }
        }
    }
    cout << res << endl;
    return 0;
}

更新日志

  • 2014年07月12日 已AC。
  • 2014年08月18日 更新了优化后的代码,提升可读性,并且运用了新的处理方法。
comments powered by Disqus