874. 模拟行走机器人
874. 模拟行走机器人 (Medium)
机器人在一个无限大小的 XY 网格平面上行走,从点 (0, 0)
处开始出发,面向北方。该机器人可以接收以下三种类型的命令 commands
:
-2
:向左转90
度-1
:向右转90
度1 <= x <= 9
:向前移动x
个单位长度
在网格上有一些格子被视为障碍物 obstacles
。第 i
个障碍物位于网格点 obstacles[i] = (xᵢ, yᵢ)
。
机器人无法走到障碍物上,它将会停留在障碍物的前一个网格方块上,但仍然可以继续尝试进行该路线的其余部分。
返回从原点到机器人所有经过的路径点(坐标为整数)的最大欧式距离的平方。(即,如果距离为 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) ,距离为 3² + 4² = 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) ,距离为 1² + 8² = 65
提示:
1 <= commands.length <= 10⁴
commands[i]
is one of the values in the list[-2,-1,1,2,3,4,5,6,7,8,9]
.0 <= obstacles.length <= 10⁴
-3 * 10⁴ <= xᵢ, yᵢ <= 3 * 10⁴
- 答案保证小于
2³¹
分析
模拟即可,思路简单,编码有点复杂.
func robotSim(commands []int, obstacles [][]int) int {
dirs := [4][2]int{{0, 1}, {1, 0}, {0, -1}, {-1, 0}}
obs := make(map[[2]int]bool, len(obstacles))
for _, o := range obstacles {
obs[[2]int{o[0], o[1]}] = true
}
dir := 0
x, y := 0, 0
res := 0
for _, c := range commands {
switch c {
case -1:
dir = (dir + 1) % 4
case -2:
dir = (dir + 3) % 4
default:
delta := dirs[dir]
for ; c > 0; c-- {
nx, ny := x+delta[0], y+delta[1]
if obs[[2]int{nx, ny}] {
break
}
x, y = nx, ny
}
res = max(res, x*x+y*y)
}
}
return res
}
func max(a, b int) int {
if a > b {
return a
}
return b
}