P1596 [USACO10OCT] Lake Counting S

这是一道很经典的搜索题目，用dfs和bfs都可以解决。

P1596 [USACO10OCT] Lake Counting S

由于近期的降雨，雨水汇集在农民约翰的田地不同的地方。我们用一个N×M(1≤N≤100,1≤M≤100) 的网格图表示。每个网格中有水（W） 或是旱地（.）。一个网格与其周围的八个网格相连，而一组相连的网格视为一个水坑。约翰想弄清楚他的田地已经形成了多少水坑。给出约翰田地的示意图，确定当中有多少水坑。

输入第 1 行：两个空格隔开的整数：N 和 M。

第 2 行到第 N+1 行：每行 M 个字符，每个字符是 W 或 .，它们表示网格图中的一排。字符之间没有空格。

输出一行，表示水坑的数量。

输入输出样例

输入 #1

10 12
W........WW.
.WWW.....WWW
....WW...WW.
.........WW.
.........W..
..W......W..
.W.W.....WW.
W.W.W.....W.
.W.W......W.
..W.......W.

输出 #1

3

说明/提示

输出细节：有三个池塘：一个在左上角，一个在右下角，还有一个在右侧。

AC代码

其实两种写法最核心的部分是一样的，解题的思路也一样，就是找到一个’W’,并将其与其连通的所有’W’都搜索一遍，并标记为’.'。需要注意的是，这里的方向是8个，即上下左右及坐上那个左下右上右下。

 for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < m; j++)
        {
            //找到水，并把他和他连通的水全都标记为.
            if (g[i][j] == 'W')
                res++, bfs(i, j);//res++, dfs(i, j);
        }
    }

• DFS

#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
#define int long long
#define endl '\n'
typedef long long ll;

const int N = 1010;
char g[N][N];
int n, m, res = 0;
int dx[8] = { -1,-1,-1,1,1,1,0,0 }, dy[8] = { 1,0,-1,1,-1,0,1,-1 };

void dfs(int a, int b)
{
    g[a][b] = '.';
    for (int i = 0; i < 8; i++)
    {
        int x = a + dx[i], y = b + dy[i];
        if (x >= 0 && x < n && y >= 0 && y < m && g[x][y] == 'W')
            dfs(x, y);
    }
}
signed main()
{
    cin >> n >> m ;

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

• BFS

#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
#define int long long
#define endl '\n'
typedef pair<int, int> PII;
typedef long long ll;
const int N = 1010;
char g[N][N];
int n, m, res = 0;
int dx[8] = { -1,-1,-1,1,1,1,0,0 }, dy[8] = { 1,0,-1,1,-1,0,1,-1 };
void bfs(int a, int b)
{
    g[a][b] = '.';
    queue<PII> q;
    q.push({ a,b });
    while(q.size())
    {
        PII u = q.front();
        q.pop();
        for (int i = 0; i < 8; i++)
        {
            int x = u.first + dx[i], y = u.second + dy[i];
            if (x >= 0 && x < n && y >= 0 && y < m && g[x][y] == 'W')
                q.push({x,y}), g[x][y] = '.';
        }
    }
}

signed main()
{
    cin >> n >> m;

    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < m; j++)
        {
            cin >> g[i][j];
        }
    }
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < m; j++)
        {
            //找到水，并把他和他连通的水全都标记为.
            if (g[i][j] == 'W')
                res++, bfs(i, j);
        }
    }
    cout << res;
    return 0;
}