题目描述
有一间长方形的房子,地上铺了红色、黑色两种颜色的正方形瓷砖。
你站在其中一块黑色的瓷砖上,只能向相邻(上下左右四个方向)的黑色瓷砖移动。
请写一个程序,计算你总共能够到达多少块黑色的瓷砖。
输入格式
输入包括多个数据集合。
每个数据集合的第一行是两个整数 W 和 H,分别表示 x 方向和 y 方向瓷砖的数量。
在接下来的 H 行中,每行包括 W 个字符。每个字符表示一块瓷砖的颜色,规则如下
1)‘.’:黑色的瓷砖;
2)‘#’:红色的瓷砖;
3)‘@’:黑色的瓷砖,并且你站在这块瓷砖上。该字符在每个数据集合中唯一出现一次。
当在一行中读入的是两个零时,表示输入结束。
输出格式
对每个数据集合,分别输出一行,显示你从初始位置出发能到达的瓷砖数(记数时包括初始位置的瓷砖)。
数据范围
1≤W,H≤20
样例
输入样例:
6 9
....#.
.....#
......
......
......
......
......
#@...#
.#..#.
0 0
输出样例:
45
算法1
(暴力枚举) $O(n^2)$
bfs 宽搜
时间复杂度 线性
参考文献 算法提高课
C++ 代码
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
#define x first
#define y second
#define read(x) scanf("%d",&x)
#define mem(a,b) memset(a,b,sizeof(a))
#define fo(i,a,b) for(int i=a;i<=b;++i)
#define go(i,a,b) for(int i=a;i>=b;--i)
const int N = 55, M = N * N;
typedef pair<int, int>PIT;
int n, m,sx,sy;
char g[N][N];
PIT q[M];
bool st[N][N];//判断重复 state
int bfs()
{
int dx[4]={0,-1,0,1},dy[4]={-1,0,1,0};
int hh=0,tt=0;
q[0]={sx,sy};
int area=0;
fo(i,0,N-1)
{
fo(j,0,N-1)
{
st[i][j]=false;
}
}
st[sx][sy]=true;
while(hh<=tt)
{
PIT t=q[hh++];
area++;
for(int i=0;i<4;i++)
{
int a=t.x+dx[i],b=t.y+dy[i];
if(a<0||a>=n||b<0||b>=m)continue;
if(st[a][b])continue;
if(g[a][b]=='#')continue;
q[++tt]={a,b};
st[a][b]=true;
}
}
return area;
}
int main()
{
while(cin>>m>>n){
if(n==0&&m==0)
return 0;
fo(i,0,n-1)
{
fo(j,0,m-1)
{
cin>>g[i][j];
}
}
fo(i,0,n-1)
{
fo(j,0,m-1)
{
if(g[i][j]=='@')
{
sx=i;
sy=j;
}
}
}
cout<<bfs()<<endl;
}
return 0;
}
算法2
(暴力枚举) $O(n^2)$
dfs 深搜
时间复杂度 线性
参考文献 算法提高课
C++ 代码
#include <iostream>
#include <cstring>
#include <algorithm>
#define endl '\n'
#define read(x) scanf_s("%d",&x)
#define mem(a,b) memset(a,b,sizeof(a))
#define fo(i,a,b) for(int i=a;i<=b;++i)
#define go(i,a,b) for(int i=a;i>=b;--i)
using namespace std;
const int N = 110;
char g[N][N];
int n,m;
bool st[N][N];
int xa, ya, xb, yb;
int dx[] = { -1,0,1,0 }, dy[] = { 0,1,0,-1 };
int dfs(int x, int y)
{
int cnt = 1;
st[x][y] = true;
fo(i, 0, 3)
{
int a = x + dx[i], b = y + dy[i];
if (a < 0 || a >= n || b < 0 || b >= m)continue;
if (g[a][b] != '.')continue;
if (st[a][b])continue;
cnt += dfs(a, b);
}
return cnt;
}
int main()
{
while (cin>>m>>n,n||m)
{
fo(i, 0, n - 1)cin >> g[i];
int x, y;
fo(i, 0, n- 1)
{
fo(j, 0, m - 1)
{
if (g[i][j] == '@')
{
x = i;
y = j;
}
}
}
mem(st,0);
cout << dfs(x, y) << endl;
}
return 0;
}