Leetcode 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) 处

解法

命令有两种类型：转向和行走，不妨分为两个部分处理，两个环节不耦合在一起。

• 转向：共有四个方向，根据在每个方向上行走时，坐标的变化，设定为 direction=[(0,1),(1,0),(0,-1),(-1,0)]。不妨以向北作为第一个方向，以变量 d 标识该方向的下标。起始为 0，则右转的方向调整为 d=(d+1)%4，左转的方向调整为 d=(d-1)%4。

• 行走：行走时根据方向，按单位调整坐标为 (p_x+direction[d][0], p_y+direction[d][1])。在向下一步移动时需要判断下一步所在位置是否为障碍点。

class Solution:
    def robotSim(self, commands: List[int], obstacles: List[List[int]]) -> int:
        p_x,p_y,ret=0,0,0
        obs_set=set([(i,j) for i,j in obstacles])
        d,direction=0,[(0,1),(1,0),(0,-1),(-1,0)]
        for c in commands:
            if c<0:
                #turn direction
                if c==-1:
                    d=(d+1)%4
                else:
                    d=(d-1)%4
            else:
                #forward
                while c>0:
                    if (p_x+direction[d][0],p_y+direction[d][1]) in obs_set:
                        break
                    p_x,p_y=p_x+direction[d][0],p_y+direction[d][1]
                    ret,c=max(ret,p_x**2+p_y**2),c-1
        return ret