AcWing 3668. 机器人走迷宫

AcWing 3668. 机器人走迷宫

有一个愚蠢的机器人走进一个 w×h的迷宫，迷宫里有空地和陷阱。他想要访问迷宫的每个方格，但是它很笨，只会按照指令的方向走。当机器人不能走的时候，也就是下一步会遇到陷阱、迷宫边界或者访问过的格子时，它会向右转 90度(顺时针旋转 90度，不能访问已经访问过的方格，且在原地只转一次，移动后可获得又一次旋转机会)。请问这个机器人最多可以经过多少个方格。

输入格式
输入包含多组测试数据。每组数据第一行包含两个整数 w,h。

接下来 w行每行有 h个字符用于描述这个迷宫。其中 . 表示空地，即为可以走的地方。* 表示陷阱，即为不能走的地方。其中还有且仅有一个英文字母，表示机器人的出发点，字母只有 U，D，L，R 四种。分别表示机器人的初始指令是向上，向下，向左，向右。

输出格式
每组数据，输出一行，一个整数，即机器人一共经过多少个方格。

数据范围
1≤w,h≤10

#include <iostream>
using namespace std;


/*
思路：
上 右 下 左 按照这个顺序来转%4即可，sta标志位表示当前机器人面向哪个方向
dfs实现，外加一个sta标志位表示是否可以转向 
*/

string map[15];
// char map[15][15] = {};
int ans = 0;
int done[15][15] = {};
int fx, sta = 0;
int n, m;

void init() {
    ans = 0;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            done[i][j] = 0;
        }
    }
}


void setFx(char c) {
    switch (c) {
    case 'U':
        fx = 0; break;
    case 'R':
        fx = 1; break;
    case 'D':
        fx = 2; break;
    case 'L':
        fx = 3; break;
    default:
        break;
    }
}

int dfs(int x, int y) {
    if ((x >= n || x < 0 || y >= m || y < 0)
        || map[x][y] == '*'
        || done[x][y]) {
        if (sta) {
            fx = (fx + 1) % 4; // 转向
            sta = 0;
            return 1; 
        }
        else return 0; // 判断是否转向过
    }
        
    done[x][y] = 1;
    sta = 1;
    ans++;
    int temp = 1;
    while (temp) {
        if (fx == 0) { temp = dfs(x - 1, y); }
        else if (fx == 1) { temp = dfs(x, y + 1); }
        else if (fx == 2) { temp = dfs(x + 1, y); }
        else if (fx == 3) { temp = dfs(x, y - 1); }
    }
    return 0;

}

int main() {
    while (cin >> n >> m) {
        init();
        int a = 0, b = 0;
        for (int i = 0; i < n; i++) {
            cin >> map[i];
            for (int j = 0; j < m; j++) {            
                if (map[i][j] >= 'A' && map[i][j] <= 'Z') a = i, b = j; // 起点
            }
        }
        setFx(map[a][b]);
        dfs(a, b);
        cout << ans << endl;
    }

    return 0;
}