【DFS迷宫问题】

题目来源：百练 红与黑

2816:红与黑
查看 提交 统计 提示 提问
总时间限制: 1000ms 内存限制: 65536kB
描述
有一间长方形的房子，地上铺了红色、黑色两种颜色的正方形瓷砖。你站在其中一块黑色的瓷砖上，只能向相邻的黑色瓷砖移动。请写一个程序，计算你总共能够到达多少块黑色的瓷砖。
输入
包括多个数据集合。每个数据集合的第一行是两个整数W和H，分别表示x方向和y方向瓷砖的数量。W和H都不超过20。在接下来的H行中，每行包括W个字符。每个字符表示一块瓷砖的颜色，规则如下
1）‘.’：黑色的瓷砖；
2）‘#’：白色的瓷砖；
3）‘@’：黑色的瓷砖，并且你站在这块瓷砖上。该字符在每个数据集合中唯一出现一次。
当在一行中读入的是两个零时，表示输入结束。
输出
对每个数据集合，分别输出一行，显示你从初始位置出发能到达的瓷砖数(记数时包括初始位置的瓷砖)。

样例输入
6 9
…#.
…#
…
…
…
…
…
#@…#
.#…#.
0 0
样例输出
45

#include <iostream>
#include <cstring>
using namespace std;
char maze[22][22];
int w, h;
int sx, sy;
bool vis[22][22];
int dir[4][2] = {{-1, 0}, {0, 1}, {1, 0}, {0, -1}};
int ans = 0;
bool judge(int x, int y)
{
    if (x < 0 || y < 0 || x >= h || y >= w || vis[x][y] == true || maze[x][y] == '#')
    {
        return false;
    }
    return true;
}

void DFS(int x, int y)
{
    vis[x][y] = true;
    ans++;
    for (int i = 0; i < 4; i++)
    {
        int nx = x + dir[i][0];
        int ny = y + dir[i][1];
        if (judge(nx, ny))
        {
            DFS(nx, ny);
        }
    }
}

int main()
{
    while (cin >> w >> h)
    {
        if (w == 0 && h == 0)
        {
            break;
        }
        for (int i = 0; i < h; i++)
        {
            for (int j = 0; j < w; j++)
            {
                cin >> maze[i][j];
                if (maze[i][j] == '@')
                {
                    sx = i;
                    sy = j;
                }
            }
        }
        memset(vis, 0, sizeof(vis));
        DFS(sx, sy);
        cout << ans << endl;
        ans = 0;
    }
    return 0;
}