题目地址(874. 模拟行走机器人)

题目地址(874. 模拟行走机器人)

https://leetcode.cn/problems/walking-robot-simulation/

题目描述

机器人在一个无限大小的 XY 网格平面上行走，从点 (0, 0) 处开始出发，面向北方。该机器人可以接收以下三种类型的命令 commands ：

-2 ：向左转 90 度
-1 ：向右转 90 度
1 <= x <= 9 ：向前移动 x 个单位长度

在网格上有一些格子被视为障碍物 obstacles 。第 i 个障碍物位于网格点  obstacles[i] = (xi, yi) 。

机器人无法走到障碍物上，它将会停留在障碍物的前一个网格方块上，但仍然可以继续尝试进行该路线的其余部分。

返回从原点到机器人所有经过的路径点（坐标为整数）的最大欧式距离的平方。（即，如果距离为 5 ，则返回 25 ）

 

注意：

北表示 +Y 方向。
东表示 +X 方向。
南表示 -Y 方向。
西表示 -X 方向。

 

示例 1：

输入：commands = [4,-1,3], obstacles = []
输出：25
解释：
机器人开始位于 (0, 0)：
1. 向北移动 4 个单位，到达 (0, 4)
2. 右转
3. 向东移动 3 个单位，到达 (3, 4)
距离原点最远的是 (3, 4) ，距离为 32 + 42 = 25

示例 2：

输入：commands = [4,-1,4,-2,4], obstacles = [[2,4]]
输出：65
解释：机器人开始位于 (0, 0)：
1. 向北移动 4 个单位，到达 (0, 4)
2. 右转
3. 向东移动 1 个单位，然后被位于 (2, 4) 的障碍物阻挡，机器人停在 (1, 4)
4. 左转
5. 向北走 4 个单位，到达 (1, 8)
距离原点最远的是 (1, 8) ，距离为 12 + 82 = 65

 

提示：

1 <= commands.length <= 104
commands[i] is one of the values in the list [-2,-1,1,2,3,4,5,6,7,8,9].
0 <= obstacles.length <= 104
-3 * 104 <= xi, yi <= 3 * 104
答案保证小于 231

前置知识

公司

• 暂无

思路

关键点

代码

• 语言支持：Python3

Python3 Code:

class Solution:
    def robotSim(self, commands: List[int], obstacles: List[List[int]]) -> int:
        # 使用一个字典存储某个方向对应的[x方向移动，y方向移动，当前方向的左侧，当前方向的右侧]
        direction = {
            'up':[0,1,'left','right'],
            'down':[0,-1,'right','left'],
            'left':[-1,0,'down','up'],
            'right':[1,0,'up','down']
        }

        """
        采用模拟方法进行行走
        记录方向（上、下、左、右）和距离
        行走过程中需要考虑障碍物的存在
        每次行走完记录距离
        """
        x,y = 0,0
        dir = 'up'
        res = 0
        obstacles = set(map(tuple,obstacles))
        for command in commands:
            if command>0:
                for step in range(command):
                    if (x+direction[dir][0],y+direction[dir][1]) not in obstacles:
                        x += direction[dir][0]
                        y += direction[dir][1]
                        res = max(res,x**2+y**2)
                    else:
                        break 
            else:
                if command == -1:
                    dir = direction[dir][3] # 右转
                else:
                    dir = direction[dir][2] # 左转
        return res 

复杂度分析

令 n 为数组长度。

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$