leetcode:模拟行走机器人

题目

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/walking-robot-simulation

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

解法

func robotSim(commands []int, obstacles [][]int) int {
    var result int

    dx := []int{0, 1, 0, -1}
    dy := []int{1, 0, -1, 0}

    xy := make(map[int]int, len(obstacles))

    var x, y, di int
    for _, obstacleValue := range obstacles {
        xy[obstacleValue[0]] = obstacleValue[1]
    }

    for _, commandValue := range commands {
        if commandValue == -2 {
            di = (di + 3) % 4
        } else if commandValue == -1 {
            di = (di + 1) % 4
        } else {
            for i := 0; i < commandValue; i++ {
                x = x + dx[di]
                y = y + dy[di]

                if xy[x] == y {
                    x = x - dx[di]
                    y = y - dy[di]
                    break
                }
            }
        }
    }

    result = x*x + y*y
    return result
}