题目描述
有一间长方形的房子,地上铺了红色、黑色两种颜色的正方形瓷砖。你站在其中一块黑色的瓷砖上,只能向相邻的黑色瓷砖移动。请写一个程序,计算你总共能够到达多少块黑色的瓷砖。
输入
包括多个数据集合。每个数据集合的第一行是两个整数W和H,分别表示x方向和y方向瓷砖的数量。W和H都不超过20。在接下来的H行中,每行包括W个字符。每个字符表示一块瓷砖的颜色,规则如下
1)‘.’:黑色的瓷砖;
2)‘#’:红色的瓷砖;(原网站上这里写成了白色)
3)‘@’:黑色的瓷砖,并且你站在这块瓷砖上。该字符在每个数据集合中唯一出现一次。
当在一行中读入的是两个零时,表示输入结束。
输出
对每个数据集合,分别输出一行,显示你从初始位置出发能到达的瓷砖数(记数时包括初始位置的瓷砖)。
样例输入
6 9
....#.
.....#
......
......
......
......
......
#@...#
.#..#.
0 0
样例输出
45
代码
这道题是比较简单的DFS问题,按照老师给的DFS结构,找清楚遍历的条件就可以了。
#include <iostream>
#include <cstring>
#include <string.h>
using namespace std;
int W, H; int sum = 0;
char tfloor[21][21]; //存放地板数据
char label[21][21]; //标记是否走过
void dfs(int m, int n)
{
if (label[m][n] == 1) //m为行,n为列
return;
//如果这块砖没有走过,那么现在走过了
label[m][n] = 1; sum += 1;
//如果没到上边界且上面一个是'.',就一直往上走。其余三个换了个方向也是同理
if (m != 0 && tfloor[m - 1][n] == '.') dfs(m - 1, n);
if (m != H - 1 && tfloor[m + 1][n] == '.') dfs(m + 1, n);
if (n != 0 && tfloor[m][n - 1] == '.') dfs(m, n - 1);
if (n != W - 1 && tfloor[m][n + 1] == '.') dfs(m, n + 1);
}
int main()
{
while (cin >> W >> H)
{
if(W==0 && H==0)
break;
int bw = 0, bh = 0; //起点位置
for (int i = 0; i < H; i++) {
for (int j = 0; j < W; j++) {
cin >> tfloor[i][j];
if (tfloor[i][j] == '@') {
//找到起点
bw = i; bh = j;
tfloor[i][j] = '.';
}
}
}
memset(label, 0, sizeof(label));//将所有标记置零
dfs(bw, bh);
cout << sum << endl;
sum = 0;
}
return 0;
}