红与黑(BFS解法)

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

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

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

输入格式

输入包括多个数据集合。

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

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

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

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

输出格式

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

数据范围

1≤W,H≤20

输入样例：

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

输出样例：

45

C++代码

#include<bits/stdc++.h>
#define pb push_back
#define fi first
#define se second
using namespace std;
using PII = pair<int, int>;

const int N = 30;
char g[N][N];
bool used[N][N];
int W, H;
queue<PII> q;

void init(){
	memset(used, false, sizeof(used));
	
	for(int i = 1; i <= H; i ++)
		for(int j = 1; j <= W; j ++)
			if(g[i][j] == '@'){
				q.push({i, j});
				used[i][j] = true;
				break;
			}
}

int main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	
	while(cin >> W >> H){
		if(! W && ! H) break;
		
		for(int i = 1; i <= H; i ++)
			for(int j = 1; j <= W; j ++)
				cin >> g[i][j];
		
		init();
		
		//BFS
		int res = 0;
		while(q.size()){
				auto t = q.front();
				res ++;
				q.pop();
				
				int dx[4] = {-1, 1, 0, 0}, dy[4] = {0, 0, 1, -1};
				for(int i = 0; i < 4; i ++){
					int x = t.fi + dx[i], y = t.se + dy[i];
					if(x >= 1 && x <= H && y >= 1 && y <= W && ! used[x][y] && g[x][y] == '.'){
						q.push({x, y});
						used[x][y] = true;
					}		
				}	
		}
			
		cout << res << '\n';
	}
		
	return 0;
}