[백준/Python] 13901 로봇

 

📌문제 탐색하기

• 로봇은 사용자가 지정한 방향을 일직선으로 움직이며, 이동 중 벽, 방문한 지역, 장애물을 만날 경우 사용자가 지정한 다음 방향으로 움직인다. 사용자가 지정한 다음 방향이 없다면 맨 처음 방향으로 돌아가서 앞의 과정을 반복한다.
• 로봇이 움직일 수 없는 경우 동작을 멈춘다.
• 방의 크기는 R * C(3 ≤ R,C ≤ 1,000), 장애물의 개수는 k(0 ≤ k ≤ 1,000)
• 로봇이 멈추는 위치를 출력하기

문제의 지시사항을 따라 로봇을 이동시키면서, 로봇이 더 이상 이동할 수 없을 때의 위치를 출력하면 된다.

 

---

📌코드 설계하기

1. 방의 크기 R, C를 입력받아 RxC 크기의 이차원 리스트 room을 만들고, 모든 요소를 0으로 초기화한다.
2. 장애물의 크기 k를 입력받고, 각 장애물의 위치를 입력받아 room의 해당 위치에 1을 저장한다.
3. 로봇의 시작 위치를 sr, sc로 입력받는다.
4. 로봇의 이동방향 순서를 저장할 리스트 direction_order에 이동 방향 순서를 입력받는다.
5. 로봇의 시작 위치에서부터 이동방향 순서와 정해진 규칙에 따라 로봇을 이동시키면서, 방문한 위치에는 방문 이력을 기록하기 위해 3을 저장한다.
6. 더 이상 로봇이 움직일 수 없을 때, 마지막 위치를 출력한다.

 

---

📌정답 코드

R, C = map(int, input().split())
room = [[0]*C for _ in range(R)]

k = int(input())
for _ in range(k):
    br, bc = map(int, input().split())
    room[br][bc] = 1

r, c = map(int, input().split())
directions = [(-1,0),(1,0),(0,-1),(0,1)]
direction_order = [directions[order-1] for order in map(int, input().split())]

room[r][c] = 3
cur_direction = 0

while True:
    for i in range(4):
        dr, dc = direction_order[cur_direction]
        nr, nc = r+dr, c+dc
        
        if 0 <= nr < R and 0 <= nc < C and not room[nr][nc]:
            r, c = nr, nc
            room[r][c] = 3
            break
        cur_direction = (cur_direction + 1) % 4
    else:
        print(r, c)
        break
else:
    print(r, c)