【Python】【难度：简单】Leetcode 874. 模拟行走机器人

机器人在一个无限大小的网格上行走，从点 (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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/walking-robot-simulation
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

class Solution(object):
    def robotSim(self, commands, obstacles):
        """
        :type commands: List[int]
        :type obstacles: List[List[int]]
        :rtype: int
        """
        obstacles=[tuple(i) for i in obstacles]
        obstacles=set(obstacles)

        var=[0,0]
        direct=1
        #1 向上
        #2 向右
        #3 向下
        #4 向左
        res=0
        for i in commands:
            if i>=0:
                tmp=i
                if direct==1:
                    while i:
                        if (var[0],var[1]+1) not in obstacles:
                            var[1]+=1
                        else:
                            break
                        i-=1
                elif direct==2:
                    while i:
                        if (var[0]+1,var[1]) not in obstacles:
                            var[0]+=1
                        else:
                            break
                        i-=1
                elif direct==3:
                    while i:
                        if (var[0],var[1]-1) not in obstacles:
                            var[1]-=1
                        else:
                            break
                        i-=1
                elif direct==4:
                    while i:
                        if (var[0]-1,var[1]) not in obstacles:
                            var[0]-=1
                        else:
                            break
                        i-=1
            else:
                if i==-1:
                    direct=(direct+1) if direct+1<=4 else 1
                if i==-2:
                    direct=direct-1 if direct-1>0 else 4

            res=max(res,var[0]**2+var[1]**2)

        return res

 

执行结果：

通过

显示详情

执行用时 :340 ms, 在所有 Python 提交中击败了93.92%的用户

内存消耗 :18.8 MB, 在所有 Python 提交中击败了100.00%的用户