总结:
对于数组的遍历问题,有个很重要的思想就是辅助列表(记录行走方向)
在这个题目中也是需要辅助列表的
最开始的fun1的思路很简单
然后fun2开始考虑障碍物问题
我最开始的想法很简单,我不想一步步走,走一步查一步(因为我觉得这样很费时间)所以我就采取了一步走到位的策略,但是这样搞了之后,我就必须将元素遍历一遍,那么每一次一个遍历,o(nm)的时间复杂度
我最开始觉得这样很好,但我忘记了字典,集合的查找性质(接近于o(1)的时间复杂度)
所以我反而更麻烦了
这就是fun3为什么可以通过而fun2为什么不可以通过的原因
我是查找了更多次,但每一次的时间更短
不考虑障碍物的代码
def fun1(commands,obstacles):
'''
思路:
创建辅助矩阵[(-1,0),(0,-1),(0,1),(1,0)]
向北 向西 向东 向南
:param commands:
:param obstacles:
:return:
'''
fuzhu=[(0,1),(-1,0),(1,0),(0,-1)]
x,y=0,0 #初始坐标
zhuangtai=0 #当前朝向
for cur in commands:
if 1<=cur<=9:
x+=fuzhu[zhuangtai][0]*cur
y+=fuzhu[zhuangtai][1]*cur
elif cur==-1:
if zhuangtai==0:
zhuangtai=2
elif zhuangtai==1:
zhuangtai=0
elif zhuangtai==2:
zhuangtai=3
else:
zhuangtai=1
else:
if zhuangtai==0:
zhuangtai=1
elif zhuangtai==1:
zhuangtai=3
elif zhuangtai==2:
zhuangtai=0
else:
zhuangtai=2
return x**2+y**2
考虑了障碍物因为时间复杂度过高而舍弃的代码
def fun2(commands,obstacles):#通过了46个最后几个估计都会超时
'''
上面的代码并没有考虑障碍物的问题
那我应该怎么解决障碍物的问题?
向上移动,应该找比他大的
向下移动,应该找比他小的
:param commands:
:param obstacles:
:return:
'''
fuzhu=[(0,1),(-1,0),(1,0),(0,-1)]
x,y=0,0 #初始坐标
zhuangtai=0 #当前朝向
max_=0
for cur in commands:
if 1<=cur<=9:
x_=x+fuzhu[zhuangtai][0]*cur
y_=y+fuzhu[zhuangtai][1]*cur
if zhuangtai==0:
#找有没有纵坐标比他大的
for i in obstacles:
if i[0]==x and y<i[1]<=y_:
y_=i[1]-1 #说明上移过程中会碰到障碍物
elif zhuangtai==1:
for i in obstacles:
if x_<=i[0]<x and i[1]==y:
x_=i[0]+1
elif zhuangtai==2:
for i in obstacles:
if x<i[0]<=x_ and i[1]==y:
x_=i[0]-1
else:
for i in obstacles:
if i[0]==x and y_<=i[1]<y:
y_=i[1]+1
x,y=x_,y_
ddd=x**2+y**2
if max_<ddd:
max_=ddd
elif cur==-1:
if zhuangtai==0:
zhuangtai=2
elif zhuangtai==1:
zhuangtai=0
elif zhuangtai==2:
zhuangtai=3
else:
zhuangtai=1
else:
if zhuangtai==0:
zhuangtai=1
elif zhuangtai==1:
zhuangtai=3
elif zhuangtai==2:
zhuangtai=0
else:
zhuangtai=2
return max_
print(fun2([-2,8,3,7,-1],[[-4,-1],[1,-1],[1,4],[5,0],[4,5],[-2,-1],[2,-5],[5,1],[-3,-1],[5,-3]]))
最开始觉得特别蠢,但又是唯一通过的代码
def fun3(commands,obstacles):
'''
fun2的代码运行没有通过
然后我想了一下,觉得应该问题出现在这里
假如我现在是(0,0) 我一会要变成(-8,0) 但是如果有一个(5,0)在哪里其实他没有当道
当时我还是会以为他当到了,那么我可以解决吗,如何判断我现在在往哪里移动(设定最开始的坐标)
ok,把fun2的代码修改了,现在的问题是超出时间限制
我看了一下答案代码,他甚至是每一次都进行一遍查找,但是还是可以通过,为什么呢?
:param commands:
:param obstacles:
:return:
'''
fuzhu = [(0, 1), (-1, 0), (1, 0), (0, -1)]
x, y = 0, 0 # 初始坐标
zhuangtai = 0 # 当前朝向
max_ = 0
ls=set(map(tuple,obstacles))
for cur in commands:
if 1<=cur<=9:
for i in range(cur):
if (x+fuzhu[zhuangtai][0],y+fuzhu[zhuangtai][1]) not in ls:
x +=fuzhu[zhuangtai][0]
y +=fuzhu[zhuangtai][1]
else:
break
ddd = x ** 2 + y ** 2
if max_ < ddd:
max_ = ddd
elif cur == -1:
if zhuangtai == 0:
zhuangtai = 2
elif zhuangtai == 1:
zhuangtai = 0
elif zhuangtai == 2:
zhuangtai = 3
else:
zhuangtai = 1
else:
if zhuangtai == 0:
zhuangtai = 1
elif zhuangtai == 1:
zhuangtai = 3
elif zhuangtai == 2:
zhuangtai = 0
else:
zhuangtai = 2
return max_