洛谷 P11228：[CSP-J 2024 T2] 地图探险 ← 搜索算法

P11228 [CSP-J 2024] 地图探险 - 洛谷

这道题今天考试考了，作为了附加题，只对了一个样例www

但是满分200分，我考了125分，非常满意

那我们来看看是怎嘛个事儿

题目描述

小 A 打算前往一片丛林去探险。丛林的地理环境十分复杂，为了防止迷路，他先派遣了一个机器人前去探路。

丛林的地图可以用一个 n 行 m 列的字符表来表示。我们将第 i 行第 j 列的位置的坐标记作 (i,j)(1≤i≤n，1≤j≤m)。如果这个位置的字符为 x，即代表这个位置上有障碍，不可通过。反之，若这个位置的字符为 .，即代表这个位置是一片空地，可以通过。

这个机器人的状态由位置和朝向两部分组成。其中位置由坐标 (x,y)(1≤x≤n，1≤y≤m) 刻画，它表示机器人处在地图上第 x 行第 y 列的位置。而朝向用一个 0∼3 的整数 d 表示，其中 d=0 代表向东，d=1 代表向南，d=2 代表向西，d=3 代表向北。

初始时，机器人的位置为 (x0​,y0​)，朝向为 d0​。保证初始时机器人所在的位置为空地。接下来机器人将要进行 k 次操作。每一步，机器人将按照如下的模式操作：

1. 假设机器人当前处在的位置为 (x,y)，朝向为 d。则它的方向上的下一步的位置 (x′,y′) 定义如下：若 d=0，则令 (x′,y′)=(x,y+1)，若 d=1，则令 (x′,y′)=(x+1,y)，若 d=2，则令 (x′,y′)=(x,y−1)，若 d=3，则令 (x′,y′)=(x−1,y)。

2. 接下来，机器人判断它下一步的位置是否在地图内，且是否为空地。具体地说，它判断 (x′,y′) 是否满足 1≤x′≤n,1≤y′≤m，且 (x′,y′) 位置上是空地。如果条件成立，则机器人会向前走一步。它新的位置变为 (x′,y′)，且朝向不变。如果条件不成立，则它会执行“向右转”操作。也就是说，令 d′=(d+1)mod4（即 d+1 除以 4 的余数），且它所处的位置保持不变，但朝向由 d 变为 d′。

小 A 想要知道，在机器人执行完 k 步操作之后，地图上所有被机器人经过的位置（包括起始位置）有几个。

输入格式

本题有多组测试数据。

输入的第一行包含一个正整数 T，表示数据组数。

接下来包含 T 组数据，每组数据的格式如下：

第一行包含三个正整数 n,m,k。其中 n,m 表示地图的行数和列数，k 表示机器人执行操作的次数。

第二行包含两个正整数 x0​,y0​ 和一个非负整数 d0​。

接下来 n 行，每行包含一个长度为 m 的字符串。保证字符串中只包含 x 和 . 两个字符。其中，第 x 行的字符串的第 y 个字符代表的位置为 (x,y)。这个位置是 x 即代表它是障碍，否则代表它是空地。数据保证机器人初始时所在的位置为空地。

输出格式

对于每组数据：输出一行包含一个正整数，表示地图上所有被机器人经过的位置（包括起始位置）的个数。

输入输出样例

输入 #1

2
1 5 4
1 1 2
....x
5 5 20
1 1 0
.....
.xxx.
.x.x.
..xx.
x....

输出 #1

3
13

说明/提示

【样例 1 解释】

该样例包含两组数据。对第一组数据，机器人的状态以如下方式变化：

1. 初始时，机器人位于位置 (1,1)，方向朝西（用数字 2 代表）。
2. 第一步，机器人发现它下一步的位置 (1,0) 不在地图内，因此，它会执行“向右转”操作。此时，它的位置仍然为 (1,1)，但方向朝北（用数字 3 代表）。
3. 第二步，机器人发现它下一步的位置 (0,1) 不在地图内，因此，它仍然会执行“向右转”操作。此时，它的位置仍然为 (1,1)，但方向朝东（用数字 0 代表）。
4. 第三步，机器人发现它下一步的位置 (1,2) 在地图内，且为空地。因此，它会向东走一步。此时，它的位置变为 (1,2)，方向仍然朝东。
5. 第四步，机器人发现它下一步的位置 (1,3) 在地图内，且为空地。因此，它会向东走一步。此时，它的位置变为 (1,3)，方向仍然朝东。

因此，四步之后，机器人经过的位置有三个，分别为 (1,1),(1,2),(1,3)。

对第二组数据，机器人依次执行的操作指令为：向东走到 (1,2)，向东走到 (1,3)，向东走到 (1,4)，向东走到 (1,5)，向右转，向南走到 (2,5)，向南走到 (3,5)，向南走到 (4,5)，向南走到 (5,5)，向右转，向西走到 (5,4)，向西走到 (5,3)，向西走到 (5,2)，向右转，向北走到 (4,2)，向右转，向右转，向南走到 (5,2)，向右转，向右转。

【数据范围】

对于所有测试数据，保证：1≤T≤5，1≤n,m≤103，1≤k≤106，1≤x0​≤n，1≤y0​≤m，0≤d0​≤3，且机器人的起始位置为空地。

问题分析
机器人需要在地图上按照特定规则移动 k 次，我们需要记录它经过的所有位置（包括起始位置）的数量。机器人的移动规则是：尝试向当前方向前进，如果不能前进（遇到障碍或边界），则向右转。

代码关键逻辑说明
方向处理：使用 dx 和 dy 数组表示四个方向（东、南、西、北）的坐标变化，简化了方向计算。
状态记录：st 数组用于记录机器人经过的位置，避免重复计数。
移动逻辑：
每次操作先计算下一步位置
判断该位置是否合法（在地图内且为空地）
若合法则移动过去，若为新位置则更新计数
若不合法则向右转（方向加 1 模 4）
时间复杂度：O (k)，其中 k 是操作次数，最多为 1e6，完全在可接受范围内。
空间复杂度：O (n*m)，用于存储地图和位置状态，n 和 m 最大为 1e3，所以空间也足够。
代码成功通过了样例测试，对于第一组数据输出 3，第二组数据输出 13，与预期结果一致。

AC代码解析

#include <bits/stdc++.h>
using namespace std;
 
const int maxn=1e3+5;
char a[maxn][maxn];  // 存储地图信息
int st[maxn][maxn];  // 记录位置是否被经过
// 方向数组：东、南、西、北对应的坐标变化
int dx[]= {0,1,0,-1};
int dy[]= {1,0,-1,0};
int n,m,k;  // 地图大小和操作次数
int x,y,d;  // 机器人当前位置和方向
 
int main() {
    int T;
    cin>>T;  // 测试数据组数
    while(T--) {
        // 初始化数组
        memset(a,0,sizeof a);
        memset(st,0,sizeof st);
        cin>>n>>m>>k;
        cin>>x>>y>>d;
        // 读取地图
        for(int i=1; i<=n; i++) {
            for(int j=1; j<=m; j++) {
                cin>>a[i][j];
            }
        }
 
        int cnt=1;  // 计数，初始位置算一个
        st[x][y]=1;  // 标记起始位置已经过
        
        // 执行k次操作
        while(k--) {
            // 计算下一步位置
            int tx=x+dx[d],ty=y+dy[d];
            // 判断是否可以前进
            if(tx>=1 && tx<=n && ty>=1 && ty<=m && a[tx][ty]=='.') {
                x=tx,y=ty;  // 前进到新位置
                // 如果是新位置，计数加1并标记
                if(!st[x][y]) {
                    cnt++;
                    st[x][y]=1;
                }
            } else {
                d=(d+1)%4;  // 不能前进则向右转
            }
        }
 
        cout<<cnt<<endl;  // 输出经过的位置总数
    }
 
    return 0;
}