[AcWing] 紅與黑

紅與黑

題目

有一間長方形的房子，地上鋪了紅色、黑色兩種顏色的正方形瓷磚。

你站在其中一塊黑色的瓷磚上，只能向相鄰（上下左右四個方向）的黑色瓷磚移動。

請寫一個程序，計算你總共能夠到達多少塊黑色的瓷磚。

輸入格式

輸入包括多個數據集合。

每個數據集合的第一行是兩個整數 W 和 H，分別表示 x 方向和 y 方向瓷磚的數量。

在接下來的 H 行中，每行包括 W 個字符。每個字符表示一塊瓷磚的顏色，規則如下

1）‘.’：黑色的瓷磚；
2）‘#’：白色的瓷磚；
3）‘@’：黑色的瓷磚，並且你站在這塊瓷磚上。該字符在每個數據集合中唯一出現一次。

當在一行中讀入的是兩個零時，表示輸入結束。

輸出格式

對每個數據集合，分別輸出一行，顯示你從初始位置出發能到達的瓷磚數(記數時包括初始位置的瓷磚)。

數據範圍

$1≤W,H≤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;
}