[백준] 2174번: 로봇 시뮬레이션 - 파이썬

🔈 문제

가로 A(1≤A≤100), 세로 B(1≤B≤100) 크기의 땅이 있다. 이 땅 위에 로봇들이 N(1≤N≤100)개 있다.

로봇들의 초기 위치는 x좌표와 y좌표로 나타난다. 위의 그림에서 보듯 x좌표는 왼쪽부터, y좌표는 아래쪽부터 순서가 매겨진다. 또한 각 로봇은 맨 처음에 NWES 중 하나의 방향을 향해 서 있다. 초기에 서 있는 로봇들의 위치는 서로 다르다.
이러한 로봇들에 M(1≤M≤100)개의 명령을 내리려고 한다. 각각의 명령은 순차적으로 실행된다. 즉, 하나의 명령을 한 로봇에서 내렸으면, 그 명령이 완수될 때까지 그 로봇과 다른 모든 로봇에게 다른 명령을 내릴 수 없다. 각각의 로봇에 대해 수행하는 명령은 다음의 세 가지가 있다.


1. L: 로봇이 향하고 있는 방향을 기준으로 왼쪽으로 90도 회전한다.
2. R: 로봇이 향하고 있는 방향을 기준으로 오른쪽으로 90도 회전한다.
3. F: 로봇이 향하고 있는 방향을 기준으로 앞으로 한 칸 움직인다.

간혹 로봇들에게 내리는 명령이 잘못될 수도 있기 때문에, 당신은 로봇들에게 명령을 내리기 전에 한 번 시뮬레이션을 해 보면서 안전성을 검증하려 한다. 이를 도와주는 프로그램을 작성하시오.
잘못된 명령에는 다음의 두 가지가 있을 수 있다.

1. Robot X crashes into the wall: X번 로봇이 벽에 충돌하는 경우이다. 즉, 주어진 땅의 밖으로 벗어나는 경우가 된다.
2. Robot X crashes into robot Y: X번 로봇이 움직이다가 Y번 로봇에 충돌하는 경우이다.

📝입력

첫째 줄에 두 정수 A, B가 주어진다. 다음 줄에는 두 정수 N, M이 주어진다. 다음 N개의 줄에는 각 로봇의 초기 위치(x, y좌표 순) 및 방향이 주어진다. 다음 M개의 줄에는 각 명령이 명령을 내리는 순서대로 주어진다. 각각의 명령은 명령을 내리는 로봇, 명령의 종류(위에 나와 있는), 명령의 반복 회수로 나타낸다. 각 명령의 반복 회수는 1이상 100이하이다.

 

📑출력

첫째 줄에 시뮬레이션 결과를 출력한다. 문제가 없는 경우에는 OK를, 그 외의 경우에는 위의 형식대로 출력을 한다. 만약 충돌이 여러 번 발생하는 경우에는 가장 먼저 발생하는 충돌을 출력하면 된다.

 

📚 문제 풀이

 

그래프를 만들고 로봇의 위치를 표시한다. 그리고 명령오는대로 회전시키거나 이동하고 그래프 범위를 벗어나거나 로봇끼리 충돌한다면 오류 메세지를 출력하면 되는 구현 문제였다.

문제에서 까다로웠던 점은 좌표가 이전과는 조금 다른 형태, 즉 아래에서 부터 위로 숫자가 증가해서 조금은 헷갈렸던거 같다.

 

로봇의 위치와 방향을 저장할 dict를 만들어두고, 그래프에는 로봇의 번호를 적어두었다.

회전은 list에 N, W, S, E 순으로 저장해두고 인덱스 계산으로 %4 나머지 연산으로 회전방향을 정하였다.

전진 명령은 dict 자료형을 만들고 키 값은 N, W, S, E 그리고 밸류는 좌표변화를 주었다.

 

로봇이 이동하였다면 이전 위치는 지워주고 새로 이동한 곳에 로봇의 번호를 적어주었다.

 

💻 코드

import sys
input = sys.stdin.readline
 
a,b = map(int,input().split()) # 좌표 정보
n,m = map(int,input().split()) # 로봇갯수, 명령어갯수
graph = [[0]*(a+1) for _ in range(b+1)] # 그래프
robot = dict() # 로봇의 위치 그리고 방향 담을 dict
move = {'N':[1,0], 'W':[0,-1], 'S':[-1,0],'E':[0,1]}
tmp = ['N','W','S','E']
res = 'OK'
for i in range(1,n+1):
    x,y,c = map(str,input().rstrip().split())
    x,y = int(x),int(y)
    graph[y][x] = i
    robot[i] = [y,x,c]
 
for i in range(m):
    ro,code,move_cnt = map(str,input().rstrip().split())
    ro = int(ro)
    nx = robot[ro][1]
    ny = robot[ro][0]
    pos = robot[ro][2]
    move_cnt = int(move_cnt)
 
    if code == 'L':
        move_cnt %= 4
        now = (tmp.index(pos)+move_cnt)%4
        robot[ro][2] = tmp[now]
 
    elif code=='R':
        move_cnt %= 4
        now = (tmp.index(pos)-move_cnt)%4
        robot[ro][2] = tmp[now]
    else:
        for j in range(move_cnt):
            nx = nx + move[pos][1]
            ny = ny + move[pos][0]
            # 인덱스 벗어난다면
            if 1 > nx or nx > a or 1 > ny or ny > b:
                if res == 'OK':
                    res = f'Robot {ro} crashes into the wall'
                break
            # 이동한 자리에 이미 다른 로봇이 있다면
            if graph[ny][nx] != 0:
                if res == 'OK':
                    res = f'Robot {ro} crashes into robot {graph[ny][nx]}'
                break
        else:
            # 변경한 위치 로봇 적음
            graph[ny][nx] = ro
            # 이전에 있던 위치 지움
            graph[robot[ro][0]][robot[ro][1]] = 0
        
            # dict 도 변경
            robot[ro][0] = ny
            robot[ro][1] = nx
 
print(res)

 

🍭 시행착오

좌표실수 예를 들어 graph[x][y] 를 적어야하는데 graph[y][x] 이런 식으로 적는 실수가 있었다.

 

🔗 문제링크 로봇 시뮬레이션

2174번: 로봇 시뮬레이션

로봇