题目描述
有一间长方形的房子,地上铺了红色、黑色两种颜色的正方形瓷砖。
你站在其中一块黑色的瓷砖上,只能向相邻(上下左右四个方向)的黑色瓷砖移动。
请写一个程序,计算你总共能够到达多少块黑色的瓷砖。
输入格式
输入包括多个数据集合。
每个数据集合的第一行是两个整数 W
和 H
,分别表示 x
方向和 y
方向瓷砖的数量。
在接下来的 H
行中,每行包括 W
个字符。每个字符表示一块瓷砖的颜色,规则如下
1)‘.’:黑色的瓷砖;
2)‘#’:红色的瓷砖;
3)‘@’:黑色的瓷砖,并且你站在这块瓷砖上。该字符在每个数据集合中唯一出现一次。
当在一行中读入的是两个零时,表示输入结束。
输出格式
对每个数据集合,分别输出一行,显示你从初始位置出发能到达的瓷砖数(记数时包括初始位置的瓷砖)。
样例
6 9
....#.
.....#
......
......
......
......
......
#@...#
.#..#.
0 0
输出样例
45
算法一
(BFS) $O(n^2)$
C++ 代码
#include<iostream>
#include<queue>
#include<cstring>
using namespace std;
typedef pair<int,int> PII;
const int N=100+10;
int n,m;
char g[N][N];
int x,y;
int bfs()
{
int d[N][N];
memset(d,-1,sizeof d);
queue<PII> q;
int dx[]={-1,0,1,0},dy[]={0,-1,0,1};
q.push({x,y});
d[x][y] = 0;
int all=0;
while(q.size()){
auto t = q.front();
q.pop();
int a=t.first,b=t.second;
for(int u=0;u<4;u++){
int i=a+dx[u],j=b+dy[u];
if(i>=0&&i<n&&j>=0&&j<m&&d[i][j]==-1&&g[i][j]=='.'){
d[i][j] = 0;
q.push({i,j});
}
}
}
for(int u=0;u<n;u++){
for(int h=0;h<m;h++){
if(d[u][h]!=-1){
all++;
}
}
}
return all;
}
int main()
{
while(cin>>m>>n)
{
if(m==0||n==0) return 0;
int all=0;
for(int u=0;u<n;u++){
for(int h=0;h<m;h++){
cin>>g[u][h];
if(g[u][h]=='@'){
x=u,y=h;
}
}
}
cout<<bfs()<<endl;
}
}