LeetCode:874. Walking Robot Simulation - Python
程序员文章站
2024-03-19 08:20:46
...
问题描述:
机器人在一个无限大小的网格上行走,从点
(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) 处
提示:
0 <= commands.length <= 10000
0 <= obstacles.length <= 10000
-30000 <= obstacle[i][0] <= 30000
-30000 <= obstacle[i][1] <= 30000
- 答案保证小于
2 ^ 31
问题分析:
首先要明白两个要求,一个是遇到障碍物时,则要停在障碍物前面,不能越过;另一个是要求返回从原点到机器人的最大欧式距离
的平方
,所以并不一定是最后位置的欧氏距离。
(1)贪心方法解决。
(2)顺序执行命令,每移动一个位置就判断是否有障碍物,如果有,则不移动,进入下一个命令,其中每移动一次就更新一次最远欧氏距离,遍历结束即可。
Python3实现:
# @Time :2018/08/06
# @Author :LiuYinxing
class Solution:
def robotSim(self, commands, obstacles):
dx, dy = [0, 1, 0, -1], [1, 0, -1, 0] # 四个方向各移动一次
x = y = di = res = 0 # x, y, 表示终点位置, di 表示方向, res 表示最后结果
obstacleSet = set(map(tuple, obstacles)) # 便于查找
for cmd in commands:
if cmd == -2: # 方向 - 向左旋转90度
di = (di - 1) % 4
elif cmd == -1: # 方向 - 向右旋转90度
di = (di + 1) % 4
else:
for k in range(cmd):
if (x+dx[di], y+dy[di]) not in obstacleSet: # 如果遇到障碍物,则停止移动
x += dx[di]
y += dy[di]
res = max(res, x*x + y*y) # 更新最大欧氏距离
return res
if __name__ == '__main__':
commands, obstacles = [4, -1, 4, -2, 4], [[2, 4]]
solu = Solution()
print(solu.robotSim(commands, obstacles))
欢迎指正哦。
上一篇: 天秤 UVA839
下一篇: SSL P2700 老妹的难题