AcWing 1113. 红与黑

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

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

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

输入格式

输入包括多个数据集合。

每个数据集合的第一行是两个整数 WW 和 HH,分别表示 xx 方向和 yy 方向瓷砖的数量。

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

1)‘.’:黑色的瓷砖;
2)‘#’:白色的瓷砖;
3)‘@’:黑色的瓷砖,并且你站在这块瓷砖上。该字符在每个数据集合中唯一出现一次。

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

输出格式

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

数据范围

1≤W,H≤20

输入样例:

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

输出样例:

45

思路:深度优先搜索(DFS),使用visit数组标记起始点‘@’以及走过的黑色瓷砖 ‘ . ’ (防止重复访问),遇到‘#’表示路径不可达直接返回即可.

#include<iostream>
#include<string>
#include<memory.h>
using namespace std;
const int N=20;
char f[N][N];
bool visit[N][N];
int d[4][2]={
   
   {-1,0},{0,-1},{1,0},{0,1}};
int m,n;
void dfs(int &res,int x,int y){
    if(x<0 || x>=m || y<0 || y>=n || visit[x][y] || f[x][y]=='#') return;
    visit[x][y]=true;
    res++;
    for(int i=0;i<4;i++){
        int nx=x+d[i][0];
        int ny=y+d[i][1];
        if(nx>=0 && nx<m && ny>=0 && ny<n) {dfs(res,nx,ny);}
    }
}
int main(){
    while(cin>>n>>m){
      if(n==0 && m==0) break;
      int x,y;
      memset(visit,false,sizeof visit);
      for(int i=0;i<m;i++){
          for(int j=0;j<n;j++){
              cin>>f[i][j];
              if(f[i][j]=='@'){x=i;y=j;}
          }
      }
      int res=0;
      dfs(res,x,y);
      cout<<res<<endl;
    }
      return 0;
}

猜你喜欢

转载自blog.csdn.net/wd2ctp88/article/details/112505167