걷는 로봇 시뮬레이션

무한 XY 평면의 로봇은 북쪽을 향한 지점(0, 0)에서 시작합니다. 로봇은 다음과 같은 세 가지 가능한 유형commands의 시퀀스를 수신할 수 있습니다.

-2 : 좌회전90도.

-1 : 우회전90도.

1 <= k <= 9 : 한 번에 한 단위씩 앞으로 이동합니다k.

그리드 사각형 중 일부는 obstacles 입니다. ith 장애물이 격자점obstacles[i] = (xi, yi)에 있습니다. 로봇이 장애물에 부딪히면 대신 현재 위치에 머무르고 다음 명령으로 이동합니다.

로봇이 원점에서 얻을 수 있는 최대 유클리드 거리를 반환합니다(즉, 거리가 5 인 경우 25 반환).

메모:

북쪽은 +Y 방향을 의미합니다.

동쪽은 +X 방향을 의미합니다.

남쪽은 -Y 방향을 의미합니다.

서쪽은 -X 방향을 의미합니다.

예 1:

입력: 명령 = [4,-1,3], 장애물 = []
출력: 25
설명: 로봇은 (0, 0)에서 시작합니다.

(0, 4)까지 북쪽으로 4단위 이동합니다.

우회전하세요.

동쪽으로 3단위 이동하여 (3, 4)로 이동합니다.
로봇이 원점에서 가장 먼 지점은 (3, 4)이며 제곱은 32 + 42 = 25 단위 떨어져 있습니다.

예 2:

입력: 명령 = [4,-1,4,-2,4], 장애물 = [[2,4]]
출력: 65
설명: 로봇은 (0, 0)에서 시작합니다.

(0, 4)까지 북쪽으로 4단위 이동합니다.

우회전하세요.

동쪽으로 1유닛 이동하고 (2, 4)에 있는 장애물에 막히면 ​​로봇은 (1, 4)에 있습니다.

좌회전하세요.

(1, 8)까지 북쪽으로 4단위 이동합니다.
로봇이 원점에서 가장 먼 지점은 (1, 8)이며 제곱은 12 + 82 = 65 단위 떨어져 있습니다.

예 3:

입력: 명령 = [6,-1,-1,6], 장애물 = []
출력: 36
설명: 로봇은 (0, 0)에서 시작합니다.

(0, 6)까지 북쪽으로 6단위 이동합니다.

우회전하세요.

우회전하세요.

(0, 0)까지 남쪽으로 6단위 이동합니다.
로봇이 원점에서 가장 먼 지점은 (0, 6)이며 제곱하면 62 = 36 단위 떨어져 있습니다.

제약:

1 <= commands.length <= 104

commands[i]는 -2 , -1 또는 [1, 9] 범위의 정수입니다.

0 <= obstacles.length <= 104

-3 * 104 <= xi, yi <= 3 * 104

답은 231 미만임이 보장됩니다.

해결책:

class Solution:
    def robotSim(self, commands: List[int], obstacles: List[List[int]]) -> int:
        obstacles = set([tuple(c) for c in obstacles])
        x, y = 0, 0
        direction = 0
        mdist = 0
        for c in commands:
            if c == -2:
                direction = (direction + 3) % 4
            elif c == -1:
                direction = (direction + 1) % 4
            else:
                if direction == 0:
                    for i in range(c):
                        if (x, y + 1) not in obstacles:
                            y += 1
                elif direction == 1:
                    for i in range(c):
                        if (x + 1, y) not in obstacles:
                            x += 1
                elif direction == 2:
                    for i in range(c):
                        if (x, y - 1) not in obstacles:
                            y -= 1
                elif direction == 3:
                    for i in range(c):
                        if (x - 1, y) not in obstacles:
                            x -= 1
                mdist = max(mdist, x * x + y * y)
        return mdist