模拟行走机器人

模拟行走机器人

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

• -2 ：向左转 90 度
• -1 ：向右转 90 度

1 <= x <= 9 ：向前移动 x 个单位长度
在网格上有一些格子被视为障碍物 obstacles 。第 i 个障碍物位于网格点 obstacles[i] = (xi, yi) 。
机器人无法走到障碍物上，它将会停留在障碍物的前一个网格方块上，但仍然可以继续尝试进行该路线的其余部分。
返回从原点到机器人所有经过的路径点（坐标为整数）的最大欧式距离的平方。（即，如果距离为 5 ，则返回 25 ）

题目思路：可以直接使用模拟去解决该问题，但是模拟该问题会有两个难点，首先第一个难点是如何模拟方向，第二个难点是如何模拟障碍物。首先，方向问题，我们可以用数字去模拟方向，例如我们定义d为数值大小为方向，d=1的时候指向北，同理，当右转的时候d+=1，d=2指向东。当超过四的时候时候只需要取余即可。左转的时候我们只需要d-=1就可以，方向的正负号取余是相同的方向，因此第一个问题就可以解决。第二问题就是如何模拟障碍，我们可以看到移动的单位最大就是9,因此我们可以直接一个单位一个单位的移动，然后就判断点是否在存在障碍物。

class Solution:
    def robotSim(self, commands: List[int], obstacles: List[List[int]]) -> int:
        dirs = [[-1, 0], [0, 1], [1, 0], [0, -1]]
        px, py, d = 0, 0, 1
        mp = set([tuple(i) for i in obstacles])
        res = 0
        for c in commands:
            if c < 0:
                d += 1 if c == -1 else -1
                d %= 4
            else:
                for i in range(c):
                    if tuple([px + dirs[d][0], py + dirs[d][1]]) in mp:
                        break
                    px, py = px + dirs[d][0], py + dirs[d][1]
                    res = max(res, px * px + py * py)
        return res