我对于这道题目的思路是这样的,dx=[0,1,0,-1] dy=[1,0,-1,0] ,其中di表示机器人移动的方向,x,y表示机器人的当前坐标,(x+dx[di],y+dy[di])表示机器人在当前坐标为参考下,下一步应该到达的点。di有三个值,分别是0,1,2,3表示坐标系的北,东,南,西。这里我们可以举一个例子,commands = [4,-1,4,-2,4], obstacles = [[2,4]],可以循环commands中的元素,如果元素小于0,则表示左转或右转,如果大于0,则表示向前移动的步数,当然还要检查是否有障碍物,由于这里是一步一步移动的,所以我们只需要判断机器人下一步的落脚点是否与obstacles中的障碍物的坐标重合,如果不重合就更新机器人的坐标,否则,就不更新。很明显commands中的第一个元素是4,大于0,表示移动的步数,di的初试值是0,机器人将向正北方移动,(x+dx[0],y+dy[0])刚好是 机 器 人 下 一 步 到 达 的 位 子,每移动一步,都要判断是否与obstacles中的障碍物坐标重合,不重合,则更新坐标,x=x+dx[di],y=y+dy[di],元素的值是多少就循环多少次,走完之后,当前机器人的坐标为(0,4)。当commands中的元素小于0时,则表示当前机器人只需要改变方向而不需要移动步数,元素等于-1,即表示右转,方向di=(di+1)%4此时di的值为1。第三个元素为4,表示步数,上一步已经得出di的值为1,即机器人将向东行走,x=x+dx[di],y=y+dy[di]由于机器人在走到(1,4)的时候,下一步将会与障碍物(2,4)重合,所以后面机器人的坐标将不再更新,即此时机器人的坐标为(1,4),di=1。第四个元素为-2,代表的是方向 di=(di-1)%4,此时di=0,方向为整北。第五个元素是4,此时di=0,x=x+dx[di],y=y+dy[di]判断移动过程中不会与障碍物坐标重合,因此,循环4此(元素为4),所以此时更新坐标(1,8),返回平方后的和结果为65。下面对着代码理解一下,思路更清晰,代码我也做了部分注释,便于推理与理解。
class Solution(object):
def robotSim(self, commands, obstacles):
dx = [0, 1, 0, -1]
dy = [1, 0, -1, 0]
x = y = di = 0 #di代表方向
obstacleSet = set(map(tuple, obstacles))
ans = 0
for cmd in commands:
#dx[di],dy[di]表示向上下左右中的一个方向移动一格
if cmd == -2:
di = (di - 1) % 4
elif cmd == -1:
di = (di + 1) % 4
else:
for k in range(cmd):
#(dx[di],dy[di])表示向上下左右中的一个方向移动一格
#如果移动一格后的坐标与障碍物重合,则不更新坐标,否则更新坐标
if (x+dx[di], y+dy[di]) not in obstacleSet:
x += dx[di]
y += dy[di]
ans = max(ans, x*x + y*y)
return ans