AcWing 1113. 红与黑(DFS)

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

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

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

输入格式

输入包括多个数据集合。

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

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

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

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

输出格式

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

数据范围

1≤W,H≤20

输入样例：

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

输出样例：

45

答案：

思路：DFS的基本应用，四个方向搜索即可

#include <iostream>
#include<bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
const int N = 1e5 + 10;
const int M = 111;
using namespace std;

int dx[]={0,1,0,-1};
int dy[]={-1,0,1,0};
bool vis[25][25];
char s[21][21];
int ans;
int w,h;

void DFS(int x,int y){
    //if(x<0||x>=w||y<0||y>=h) return ;
    //cout<<x<<" "<<y<<endl;
    vis[x][y]=1;
    for(int i=0;i<4;i++){
        if(s[x+dx[i]][y+dy[i]]=='.'&&!vis[x+dx[i]][y+dy[i]]){
            ans++;
            DFS(x+dx[i],y+dy[i]);
        }
    }
}

int main()
{
    while(cin>>w>>h){
        memset(vis,0,sizeof(vis));
        if(!w&&!h) break;
        int x,y;
        for(int i=0;i<h;i++){
            for(int j=0;j<w;j++){
                cin>>s[i][j];
                if(s[i][j]=='@'){
                    x=i;
                    y=j;
                }
            }
        }
        ans=1;
        DFS(x,y);
        //cout<<x<<" "<<y<<endl;
        cout<<ans<<endl;
    }
    return 0;
}