题目描述
解题思路
先假设没有障碍的时候,就使用模拟法即可,主要是注意方向的变换
有障碍时,题目有个坑,即在起点处就有障碍,要特意排除这种情况
查找是否有障碍时,不去遍历所有障碍,而是在这条线上从小到大遍历一遍,看是否存在障碍
代码
class Solution:
def robotSim(self, commands: List[int], obstacles: List[List[int]]) -> int:
n=len(commands)
if n<1:
return 0
newobstacles=set([tuple(i) for i in obstacles])#用set存储obstacles
if (0,0) in newobstacles:#排除起点有障碍的情况,直接去除这个障碍,不影响后续操作
newobstacles.remove((0,0))
x,y=0,0
direction=[(0,1),(1,0),(0,-1),(-1,0)]#上左下右
curdirection=(0,1)#初始方向向北
res=0
for command in commands:
if command==-1:
temp=direction.index(curdirection)
curdirection=direction[(temp+1)%4]#每次向右转,方向就转向下一个状态
elif command==-2:
temp=direction[::-1].index(curdirection)
curdirection=direction[::-1][(temp+1)%4]#向左转则,与上面相反即可
else:
newx=x+command*curdirection[0]
newy=y+command*curdirection[1]#先假设没有障碍时,当前能到达的位置
if obstacles:
if newx==x:#x相等,去y轴从小到大遍历
for tempy in range(min(y,newy),max(y,newy)+1):
if (x,tempy) in newobstacles:
newy=tempy-curdirection[1]#若有障碍,则退回一步,即与原来方向相反的走一步
break;
else:
for tempx in range(min(x,newx),max(x,newx)+1):
if (tempx,y) in newobstacles:
newx=tempx-curdirection[0]
break;
x,y=newx,newy#每次有新的位置点产生时,计算欧式距离
res=max(res,x**2+y**2)
return res