[백준/1175] 배달 (Python)

Problem : https://www.acmicpc.net/problem/1175

1175번: 배달

 

Difficulty : Gold 1

 

Status : Solved

 

Time : 00:18:48

 

---

 

문제 설명

 

어제 선물을 모두 포장한 민식이는 이제 선물을 배달하려고 한다. 민식이가 선물을 배달할 곳은 이 문제를 읽는 사람들이 앉아 있는 교실이다. 교실은 직사각형모양이고, 모두 같은 크기의 정사각형 블록으로 나누어져 있다.

입력으로 교실의 지도가 주어진다. 각각의 정사각형 블록은 다음과 같이 4가지 종류가 있다.

S: 지금 민식이가 있는 곳이다. 이곳이 민식이가 배달을 시작하는 곳이고 1개만 있다.
C: 민식이가 반드시 선물을 배달해야 하는 곳이다. 이러한 블록은 정확하게 2개 있다.
#: 민식이가 갈 수 없는 곳이다.
.: 민식이가 자유롭게 지나갈 수 있는 곳이다.
민식이가 한 블록 동서남북으로 이동하는데는 1분이 걸린다. 민식이는 네가지 방향 중 하나로 이동할 수 있으며, 교실을 벗어날 수 없다. 민식이가 선물을 배달해야 하는 곳에 들어갈 때, 민식이는 그 곳에 있는 사람 모두에게 선물을 전달해야 한다. 이 상황은 동시에 일어나며, 추가적인 시간이 소요되지 않는다.

민식이는 어느 누구도 자신을 보지 않았으면 하기 때문에, 멈추지 않고 매 시간마다 방향을 바꿔야 한다. 이 말은 같은 방향으로 두 번 연속으로 이동할 수 없다는 말이다. 민식이가 선물을 모두 배달하는데 걸리는 시간의 최솟값을 구하는 프로그램을 작성하시오.

 

입력 및 출력

 

입력

 

첫째 줄에 교실의 세로 크기 N과 가로 크기 M이 주어진다. N과 M은 50보다 작거나 같은 자연수이다. 둘째 줄부터 N개의 줄에 교실의 지도가 주어진다.

 

출력

 

첫째 줄에 민식이가 선물을 모두 배달하는데 걸리는 시간의 최솟값을 출력한다. 만약 불가능 할 때는 -1을 출력한다.

 

입력 예시

 

2 3
SCC
...

 

출력 예시

 

4

 

 

---

 

풀이

 

그래프 문제. 우리는 BFS로 문제를 풀되, visited에 저장해야 할 정보를 좀 더 세분화할 필요가 있다. 방문한 C(0 <= C <= 3)의 비트마스킹 정보, 직전의 진입 방향 k (0 <= k <= 3)이 그것이다. 직전 진입 방향을 제외한 다른 방향으로 이동할 수 있어야 하며, 방문할 장소가 C인 경우 C 방문 정보를 업데이트해야 한다.

 

풀이 코드

from collections import deque
import sys
input = sys.stdin.readline
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]
MAX = float('inf')

N, M = map(int, input().split())
map_list = list()
C = list()
for i in range(N) :
  _map_list = input().strip()
  for j in range(M) :
    if _map_list[j] == 'S' :
      sx, sy = j, i
    if _map_list[j] == 'C' :
      C.append((j, i))
  map_list.append(_map_list)

q = deque()
visited = [[[[MAX]*4 for _ in range(4)] for _ in range(M)] for _ in range(N)]
for i in range(4) :
  q.append((sx, sy, i, 0, 0))
  visited[sy][sx][0][i] = 0
answer = MAX

while q :
  x, y, dir, c, dist = q.popleft()
  if c == 3 :
    answer = min(answer, dist)
    break
  for k in range(4) :
    if k == dir :
      continue
    ax, ay = x + dx[k], y + dy[k]
    if -1 < ay < N and -1 < ax < M and map_list[ay][ax] != '#' :
      if map_list[ay][ax] == 'C' :
        ac = c | 2 if (ax, ay) == C[0] else c | 1
      else :
        ac = c
      if visited[ay][ax][ac][k] > dist + 1 :
        visited[ay][ax][ac][k] = dist + 1
        q.append((ax, ay, k, ac, dist+1))

print(answer if answer < MAX else -1)

풀이 완료!