VOJ 红与黑 题解 BFS

红与黑

题目描述

有一间长方形的房子，地上铺了红色、黑色两种颜色的正方形瓷砖。你站在其中一块黑色的瓷砖上，只能向相邻的黑色瓷砖移动。请写一个程序，计算你总共能够到达多少块黑色的瓷砖。

输入描述

包括多个数据集合。每个数据集合的第一行是两个整数W和H，分别表示x方向和y方向瓷砖的数量。W和H都不超过20。在接下来的H行中，每行包括W个字符。每个字符表示一块瓷砖的颜色，规则如下:

1）‘.’：黑色的瓷砖；

2）‘#’：白色的瓷砖；

3）‘@’：黑色的瓷砖，并且你站在这块瓷砖上。该字符在每个数据集合中唯一出现一次。

当在一行中读入的是两个零时，表示输入结束。

输出描述

对每个数据集合，分别输出一行，显示你从初始位置出发能到达的瓷砖数(记数时包括初始位置的瓷砖)。

用例输入 1

6 9 
....#.
.....#
......
......
......
......
......
#@...#
.#..#.
0 0

用例输出 1

45

思路

典中典之BFS，问题转化为求包含起点在内的连通块的大小，通过BFS找连通的’ . ‘和’@'的数量即可。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
struct node // 点坐标的结构体
{
    int x, y;
};
// 四个方向
int dx[4] = {0, 0, -1, 1};
int dy[4] = {-1, 1, 0, 0};
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    int n, m;
    while (1)
    {
        cin >> m >> n; // 注意是先读入x方向的长度，别读入读反了0.0
        char a[22][22] = {0};
        bool vis[22][22] = {0}; // 是否已访问
        if (!n && !m)           // 输入为两个零时，输入结束，退出循环
            break;
        queue<node> q;
        for (int i = 1; i <= n; i++)
        {
            for (int j = 1; j <= m; j++)
            {
                cin >> a[i][j];
                if (a[i][j] == '@') // 将起点加入队列
                {
                    q.push({i, j});
                    vis[i][j] = 1; // 起点已访问
                }
            }
        }
        int ans = 1; // 将起点统计到答案里
        while (!q.empty())
        {
            int size = q.size();
            while (size--)
            {
                node tmp = q.front();
                q.pop();
                int nx, ny;
                // 遍历四个方向
                for (int i = 0; i < 4; i++)
                {
                    nx = dx[i] + tmp.x;
                    ny = dy[i] + tmp.y;
                    if (!vis[nx][ny] && a[nx][ny] == '.') // 找到一个新的黑瓷砖
                    {
                        vis[nx][ny] = 1;
                        q.push({nx, ny});
                        ans++; // 瓷砖数++
                    }
                }
            }
        }
        cout << ans << '\n';
    }
    return 0;
}