题目描述
有一间长方形的房子,地上铺了红色、黑色两种颜色的正方形瓷砖。你站在其中一块黑色的瓷砖上,只能向相邻的黑色瓷砖移动。请写一个程序,计算你总共能够到达多少块黑色的瓷砖。
输入格式
包括多个数据集合。每个数据集合的第一行是两个整数W和H,分别表示x方向和y方向瓷砖的数量。W和H都不超过20。在接下来的H行中,每行包括W个字符。每个字符表示一块瓷砖的颜色,规则如下:
- ‘.’:黑色的瓷砖;
- ‘#’:白色的瓷砖;
- ‘@’:黑色的瓷砖,并且你站在这块瓷砖上。该字符在每个数据集合中唯一出现一次;
当在一行中读入的是两个零时,表示输入结束。
输出格式
对每个数据集合,分别输出一行,显示你从初始位置出发能到达的瓷砖数(记数时包括初始位置的瓷砖)。
输入样例
6 9
…#.
…#
…
…
…
…
…
#@…#
.#…#.
0 0输出样例
45
思路
典型的搜索题,注意为避免搜索越界需要在其边界加上“篱笆”。使用BFS或者DFS搜索均可,时间复杂度O(nm)。
源程序
DFS版本
#include <bits/stdc++.h>
#define MAXN 25
using namespace std;
int n,m,sx,sy,cnt;
int mov[4][2]={{-1,0},{0,1},{1,0},{0,-1}};
char g[MAXN][MAXN];
bool used[MAXN][MAXN];
void dfs(int x,int y)
{
for(int i=0;i<4;i++){
int xx=x+mov[i][0];
int yy=y+mov[i][1];
if(g[xx][yy]=='.'&&!used[xx][yy]){
used[xx][yy]=true;
cnt++;
dfs(xx,yy);
}
}
}
int main()
{
while(1){
cin>>m>>n;
if(n==0&&m==0)break;
memset(used,false,sizeof(used));
for(int i=1;i<=n;i++) //输入数据
for(int j=1;j<=m;j++){
cin>>g[i][j];
if(g[i][j]=='@')sx=i,sy=j;
}
for(int i=0;i<=m+1;i++)g[0][i]=g[n+1][i]='#'; //建立上下篱笆
for(int i=0;i<=n+1;i++)g[i][0]=g[i][m+1]='#'; //建立左右篱笆
used[sx][sy]=true,cnt=1; //初始化
dfs(sx,sy);
cout<<cnt<<endl;
}
return 0;
}
BFS版本
#include <bits/stdc++.h>
#define MAXN 25
using namespace std;
struct Node{
int x,y;
};
int n,m,sx,sy,cnt;
int mov[4][2]={{-1,0},{0,1},{1,0},{0,-1}};
char g[MAXN][MAXN];
bool used[MAXN][MAXN];
void bfs(int x,int y)
{
queue<Node> q;
memset(used,false,sizeof(used));//初始化
used[sx][sy]=true,cnt=1;
q.push(Node{sx,sy});
while(!q.empty()){
Node tmp=q.front();
q.pop();
for(int i=0;i<4;i++){
int xx=tmp.x+mov[i][0];
int yy=tmp.y+mov[i][1];
if(g[xx][yy]=='.'&&!used[xx][yy]){
used[xx][yy]=true;
cnt++;
q.push(Node{xx,yy});
}
}
}
}
int main()
{
while(1){
cin>>m>>n;
if(n==0&&m==0)break;
for(int i=1;i<=n;i++) //输入数据
for(int j=1;j<=m;j++){
cin>>g[i][j];
if(g[i][j]=='@')sx=i,sy=j;
}
for(int i=0;i<=m+1;i++)g[0][i]=g[n+1][i]='#'; //建立上下篱笆
for(int i=0;i<=n+1;i++)g[i][0]=g[i][m+1]='#'; //建立左右篱笆
bfs(sx,sy);
cout<<cnt<<endl;
}
return 0;
}