[AcWing] 红与黑

红与黑

题目

有一间长方形的房子，地上铺了红色、黑色两种颜色的正方形瓷砖。

你站在其中一块黑色的瓷砖上，只能向相邻（上下左右四个方向）的黑色瓷砖移动。

请写一个程序，计算你总共能够到达多少块黑色的瓷砖。

输入格式

输入包括多个数据集合。

每个数据集合的第一行是两个整数 W 和 H，分别表示 x 方向和 y 方向瓷砖的数量。

在接下来的 H 行中，每行包括 W 个字符。每个字符表示一块瓷砖的颜色，规则如下

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

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

输出格式

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

数据范围

$1\le W,H\le 20$

输入样例

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

输出样例

45

题解

思路

• flood fill 算法通过 BFS 实现

#include <iostream>
#include <queue>
#include <cstring>

using namespace std;

const int N = 22;
int n, m, sx, sy, res, dx[]={0,1,0,-1}, dy[]={1,0,-1,0};
char g[N][N];
bool st[N][N];
queue<pair<int,int>> q;

void bfs() {
    st[sx][sy] = true;
    q.push({sx, sy});
    
    while (q.size()) {
        auto t = q.front();
        q.pop();
        int x = t.first, y = t.second;
        
        for (int i = 0; i < 4; i ++) {
            int a = x + dx[i], b = y + dy[i];
            if (a >= 0 && b >= 0 && a < n && b < m && !st[a][b] && g[a][b] == '.') {
                q.push({a, b});
                st[a][b] = true;
                res ++;
            }
        }
    }
}

int main () {
    while (cin >> m >> n, m || n) {
        res = 0;
        memset(st, 0, sizeof st);
        for (int i = 0; i < n; i ++) 
            for (int j = 0; j < m; j ++) {
                cin >> g[i][j];
                if (g[i][j] == '@') sx = i, sy = j;
            }
        
        bfs();
        cout << ++ res << endl;
    }
    return 0;
}