题目描述
机器人在一个无限大小的网格上行走,从点 (0, 0) 处开始出发,面向北方。该机器人可以接收以下三种类型的命令:
第 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) 处
解法
命令有两种类型:转向和行走,不妨分为两个部分处理,两个环节不耦合在一起。
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