每天一道leetcode：874.模拟行走机器人

今日份题目：

机器人在一个无限大小的 XY 网格平面上行走，从点 (0, 0) 处开始出发，面向北方。该机器人可以接收以下三种类型的命令 commands ：

-2 ：向左转 90 度 -1 ：向右转 90 度 1 <= x <= 9 ：向前移动 x 个单位长度 在网格上有一些格子被视为障碍物 obstacles 。第 i 个障碍物位于网格点 obstacles[i] = (xi, yi) 。

机器人无法走到障碍物上，它将会停留在障碍物的前一个网格方块上，但仍然可以继续尝试进行该路线的其余部分。

返回从原点到机器人所有经过的路径点（坐标为整数）的最大欧式距离的平方。（即，如果距离为 5 ，则返回 25 ）

注意：

• 北表示 +Y 方向。

• 东表示 +X 方向。

• 南表示 -Y 方向。

• 西表示 -X 方向。

示例1

输入：commands = [4,-1,3], obstacles = []
输出：25
解释：
机器人开始位于 (0, 0)：
1. 向北移动 4 个单位，到达 (0, 4)
2. 右转
3. 向东移动 3 个单位，到达 (3, 4)
距离原点最远的是 (3, 4) ，距离为 32 + 42 = 25

示例2

输入：commands = [4,-1,4,-2,4], obstacles = [[2,4]]
输出：65
解释：机器人开始位于 (0, 0)：
1. 向北移动 4 个单位，到达 (0, 4)
2. 右转
3. 向东移动 1 个单位，然后被位于 (2, 4) 的障碍物阻挡，机器人停在 (1, 4)
4. 左转
5. 向北走 4 个单位，到达 (1, 8)
距离原点最远的是 (1, 8) ，距离为 12 + 82 = 65

提示

1 <= commands.length <= 104
commands[i] is one of the values in the list [-2,-1,1,2,3,4,5,6,7,8,9].
0 <= obstacles.length <= 104
-3 * 104 <= xi, yi <= 3 * 104
答案保证小于 231

题目思路

使用unordered_set存放障碍物信息，使用count函数查找是否存在其中；使用二维数组dirs存放四个方向的信息；遍历指令数组，如果需要转向，就通过改变d中记录的数字进行转向操作，并通过取余和+4进行方向的修正，如果不需转向，就直接对x和y进行移动操作，并更新最大距离。

官方代码

class Solution 
{
public:
    int robotSim(vector<int>& commands, vector<vector<int>>& obstacles) 
    {
        int dirs[4][2] = {{-1, 0}, {0, 1}, {1, 0}, {0, -1}};//定义四个方向
        int px = 0, py = 0, d = 1;//方向初始化为1，向北行走
        unordered_set<int> mp;
        for (auto &o : obstacles) //遍历障碍物数组
        {
            mp.emplace(o[0] * 60001 + o[1]);//放入障碍物信息
        }
        int result = 0;
        for (int c : commands) //遍历指令数组
        {
            if (c < 0) //需要转向
            {
                d += c == -1 ? 1 : -1;//等于-1说明是向左转，否则是向右转
                d %= 4;//找方向
                if (d < 0) 
                {
                    d += 4;
                }
            } 
            else 
            {
                for (int i = 0; i < c; i++) 
                {
                    if (mp.count((px + dirs[d][0]) * 60001 + py + dirs[d][1])) 
                    {//count用来判断元素是否在set内，如果在，说明是障碍物，过不去
                        break;
                    }
                    //在当前方向上移动
                    px += dirs[d][0];
                    py += dirs[d][1];
                    result = max(result, px * px + py * py);//更新距离的最大值
                }
            }
        }
        return result;
    }
};

提交结果