1 Star 0 Fork 0

yuhang2__2/LeetCode-Solutions

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
walking-robot-simulation.py 815 Bytes
一键复制 编辑 原始数据 按行查看 历史
# Time: O(n + k)
# Space: O(k)
class Solution(object):
def robotSim(self, commands, obstacles):
"""
:type commands: List[int]
:type obstacles: List[List[int]]
:rtype: int
"""
directions = [(0, 1), (1, 0), (0, -1), (-1, 0)]
x, y, i = 0, 0, 0
lookup = set(map(tuple, obstacles))
result = 0
for cmd in commands:
if cmd == -2:
i = (i-1) % 4
elif cmd == -1:
i = (i+1) % 4
else:
for k in xrange(cmd):
if (x+directions[i][0], y+directions[i][1]) not in lookup:
x += directions[i][0]
y += directions[i][1]
result = max(result, x*x + y*y)
return result
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/yuhang2__2/LeetCode-Solutions.git
git@gitee.com:yuhang2__2/LeetCode-Solutions.git
yuhang2__2
LeetCode-Solutions
LeetCode-Solutions
master

搜索帮助