LeetCode：874. Walking Robot Simulation - Python

问题描述：

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

• -2：向左转 90 度
• -1：向右转 90 度
• 1 <= x <= 9：向前移动 x 个单位长度
  在网格上有一些格子被视为障碍物。

第 i 个障碍物位于网格点 (obstacles[i][0], obstacles[i][1])

如果机器人试图走到障碍物上方，那么它将停留在障碍物的前一个网格方块上，但仍然可以继续该路线的其余部分。

返回从原点到机器人的最大欧式距离的平方。
示例 1：

输入: commands = [4,-1,3], obstacles = []
输出: 25
解释: 机器人将会到达 (3, 4)

示例 2：

输入: commands = [4,-1,4,-2,4], obstacles = [[2,4]]
输出: 65
解释: 机器人在左转走到 (1, 8) 之前将被困在 (1, 4) 处

提示：

• 0 <= commands.length <= 10000
• 0 <= obstacles.length <= 10000
• -30000 <= obstacle[i][0] <= 30000
• -30000 <= obstacle[i][1] <= 30000
• 答案保证小于 2 ^ 31

问题分析：

首先要明白两个要求，一个是遇到障碍物时，则要停在障碍物前面，不能越过；另一个是要求返回从原点到机器人的最大欧式距离的平方，所以并不一定是最后位置的欧氏距离。
（1）贪心方法解决。
（2）顺序执行命令，每移动一个位置就判断是否有障碍物，如果有，则不移动，进入下一个命令，其中每移动一次就更新一次最远欧氏距离，遍历结束即可。

Python3实现：

# @Time   :2018/08/06
# @Author :LiuYinxing


class Solution:
    def robotSim(self, commands, obstacles):
        dx, dy = [0, 1, 0, -1], [1, 0, -1, 0]  # 四个方向各移动一次
        x = y = di = res = 0  # x, y, 表示终点位置， di 表示方向, res 表示最后结果
        obstacleSet = set(map(tuple, obstacles))  # 便于查找

        for cmd in commands:
            if cmd == -2:  # 方向 - 向左旋转90度
                di = (di - 1) % 4
            elif cmd == -1:  # 方向 - 向右旋转90度
                di = (di + 1) % 4
            else:
                for k in range(cmd):
                    if (x+dx[di], y+dy[di]) not in obstacleSet:  # 如果遇到障碍物，则停止移动
                        x += dx[di]
                        y += dy[di]
                        res = max(res, x*x + y*y)  # 更新最大欧氏距离
        return res


if __name__ == '__main__':
    commands, obstacles = [4, -1, 4, -2, 4],  [[2, 4]]
    solu = Solution()
    print(solu.robotSim(commands, obstacles))

欢迎指正哦。