NOI / 2.5基本算法之搜索 1818:红与黑 dfs

描述

有一间长方形的房子，地上铺了红色、黑色两种颜色的正方形瓷砖。你站在其中一块黑色的瓷砖上，只能向相邻的黑色瓷砖移动。请写一个程序，计算你总共能够到达多少块黑色的瓷砖。

输入

包括多个数据集合。每个数据集合的第一行是两个整数W和H，分别表示x方向和y方向瓷砖的数量。W和H都不超过20。在接下来的H行中，每行包括W个字符。每个字符表示一块瓷砖的颜色，规则如下
1）‘.’：黑色的瓷砖；
2）‘#’：红色的瓷砖；
3）‘@’：黑色的瓷砖，并且你站在这块瓷砖上。该字符在每个数据集合中唯一出现一次。
当在一行中读入的是两个零时，表示输入结束。

输出

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

样例输入

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

样例输出

45

#include<bits/stdc++.h>
using namespace std;
char a[21][21];
int w,h,cnt;
bool vis[21][21];
int dx[]={0,0,1,-1};
int dy[]={1,-1,0,0};

void dfs(int x,int y){
    vis[x][y]=1;//当前点标记
    cnt++;//计数
    for(int i=0;i<4;i++){
        int nx=x+dx[i];
        int ny=y+dy[i];
        if(nx<1||ny<1||nx>h||ny>w)continue;//出地图
        if(a[nx][ny]!='.')continue;//如果不是黑砖
        if(vis[nx][ny])continue;//已经在当前路径中
        dfs(nx,ny);
    }
}

int main(){
    while(true){
        cin>>w>>h;
        if(!w&&!h)break;
        int sx=-1,sy=-1;
        for(int i=1;i<=h;i++){
            cin>>(a[i]+1);
            for(int j=1;j<=w;j++){
                if(a[i][j]=='@'){
                    sx=i;
                    sy=j;
                }
            }
        }
        if(sx==-1 && sy==-1){
            cout<<0<<endl;
            continue;
        }
        cnt=0;
        //多组数据需要把数组初始化
        memset(vis,false,sizeof(vis));
        dfs(sx,sy);
        cout<<cnt<<endl;
    }
    return 0;
}