【ACWing】1113. 红与黑

题目地址:

https://www.acwing.com/problem/content/1115/

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

输入格式:
输入包括多个数据集合。每个数据集合的第一行是两个整数 W W W H H H,分别表示 x x x方向和 y y y方向瓷砖的数量。在接下来的 H H H行中,每行包括 W W W个字符。每个字符表示一块瓷砖的颜色,规则如下
(1).:黑色的瓷砖;
(2)#:红色的瓷砖;
(3)@:黑色的瓷砖,并且你站在这块瓷砖上。该字符在每个数据集合中唯一出现一次。
当在一行中读入的是两个零时,表示输入结束。

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

数据范围:
1 ≤ W , H ≤ 20 1≤W,H≤20 1W,H20

从出发点开始DFS,算一下与@处于同一 4 4 4连通块的.的个数即可。代码如下:

#include <iostream>
using namespace std;

const int N = 25, d[] = {
    
    1, 0, -1, 0, 1};
int n, m;
char g[N][N];
int res;

void dfs(int x, int y) {
    
    
    res++;
    g[x][y] = '#';
    for (int i = 0; i < 4; i++) {
    
    
        int nx = x + d[i], ny = y + d[i + 1];
        if (0 <= nx && nx < m && 0 <= ny && ny < n && g[nx][ny] == '.')
            dfs(nx, ny);
    }
}

int main() {
    
    
    while (1) {
    
    
        cin >> n >> m;
        if (n == 0 && m == 0) break;

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

    return 0;
}

每组数据时空复杂度 O ( n m ) O(nm) O(nm)

猜你喜欢

转载自blog.csdn.net/qq_46105170/article/details/114813642