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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/walking-robot-simulation/
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

# 初始化字典保存各个方向,遍历commands,根据所给数字进行具体操作并判断移动过程中是否会遇到障碍物即可
class Solution:
    def robotSim(self, commands: List[int], obstacles: List[List[int]]) -> int:
    	# 初始化字典(当前方向所对应的x值,当前方向所对应的y值,当前方向的左侧,当前方向的右侧)
        dire = {"up": [0, 1, "left", "right"],
                "down": [0, -1, "right", "left"],
                "left": [-1, 0, "down", "up"],
                "right": [1, 0, "up", "down"]}
        x, y = 0, 0  # 初始化机器人当前坐标
        dir = "up"  # 初始化初始移动方向
        res = 0  # 初始化最终结果
        obstacles = set(map(tuple, obstacles))  # 将障碍物列表转换为元组并去重,节约运算时间

        for i in commands:
        	# 若当前遍历的数字为负数则变更前进方向
            if i < 0:
                if i == -1:
                    dir = dire[dir][3]
                else:
                    dir = dire[dir][2]
            else:
                for j in range(i):  # 每移动一格便判断一次移动后的坐标是否与障碍物坐标重叠
                    if (x + dire[dir][0], y + dire[dir][1]) not in obstacles:
                        x += dire[dir][0]
                        y += dire[dir][1]
                        res = max(res, x ** 2 + y ** 2)
                    else:
                        break
                        
        return res

你可能感兴趣的:(LeetCode)