红与黑题解 【BFS】

红与黑

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

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

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

输入格式
输入包括多个数据集合。

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

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

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

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

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

数据范围
1≤W,H≤20
输入样例：

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

输出样例：

45

分析

直接用bfs就行了可以自己写队列也可以用C++的STL，只不过要记得排重（用st数组来看是否有被选中过），最后我们在结束一组数组后记得要把st的状态恢复!!!

---

代码

#include<iostream>
#include<algorithm>
#include<queue>
#include<string>
#include<string.h>
using namespace std;
typedef pair<int, int>PII;
const int N = 25;
char c[N][N];
bool st[N][N];
int startx, starty;
int dx[] = { -1,0,1,0 };
int dy[] = { 0,1,0,-1 };
int sum;
int n, m;
int bfs(int x, int y) {
	sum = 0;
	queue<PII>qq;
	qq.push({ x,y });
	st[x][y] = true;
	while (qq.size()) {
		PII t = qq.front();
		qq.pop();
		sum++;
		for (int i = 0; i < 4; i++) {
			int x1 = dx[i] + t.first;
			int y1 = dy[i] + t.second;
			if (x1 < 0 || x1 >= m || y1 < 0 || y1 >= n || c[x1][y1] != '.'||st[x1][y1]==true) continue;
			qq.push({ x1,y1 });
			st[x1][y1] = true;
		}
	}
	return sum;
}
int main() {
	
	while (1) {
		cin >> n >> m;
		if (n == 0 && m == 0)break;
		for (int i = 0; i < m; i++) {
			cin >> c[i];
		}
		for (int i = 0; i < m; i++) {
			for (int j = 0; j < n; j++) {
				if (c[i][j] == '@') {
					startx = i, starty = j;
					break;
				}
			}
		}
		cout << bfs(startx, starty) << endl;
		memset(st,false,sizeof(st));
	}
	return 0;
}

---

运行结果