题目地址:
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
1≤W,H≤20
从出发点开始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)。