# -*- coding: utf-8 -*-
# @Time    : 2019/12/28 0028 9:50
# @Author  : songzhenxi
# @Email   : songzx_2326@163.com
# @File    : LeetCode_874_1034.py
# @Software: PyCharm

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


# leetcode submit region begin(Prohibit modification and deletion)
class Solution(object):
    def robotSim(self, commands, obstacles):
        """
        题目：874.模拟行走的机器人（https://leetcode-cn.com/problems/walking-robot-simulation/submissions/）
        学号：1034（五期一班三组）
        :type commands: List[int]
        :type obstacles: List[List[int]]
        :rtype: int
        """
        i = j = mx = d = 0
        move, obstacles = [(0, 1), (-1, 0), (0, -1), (1, 0),], set(map(tuple, obstacles))
        for cmd in commands:
            if cmd == -2:
                d = (d + 1) % 4
            elif cmd == -1:
                d = (d - 1) % 4
            else:
                x, y = move[d]
                while cmd and (i + x, j + y) not in obstacles:
                    i += x
                    j += y
                    cmd -= 1
            mx = max(mx, i ** 2 + j ** 2)
        return mx
# leetcode submit region end(Prohibit modification and deletion)
