새소식

PS/백준

[백준/1938] 통나무 옮기기 (Python)

  • -

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

 

Difficulty : Gold 2

 

Status : Solved

 

Time : 00:19:40

 


 

문제 설명

 

더보기

 

가로와 세로의 길이가 같은 평지에서 벌목을 한다. 그 지형은 0과 1로 나타나 있다. 1은 아직 잘려지지 않은 나무를 나타내고 0은 아무 것도 없음을 나타낸다. 다음 지형을 보자.

위의 지형에서 길이 3인 통나무 BBB를 밀거나 회전시켜 EEE의 위치로 옮기는 작업을 하는 문제를 생각해 보자. BBB와 EEE의 위치는 임의로 주어진다. 단 문제에서 통나무의 길이는 항상 3이며 B의 개수와 E의 개수는 같다. 통나무를 움직이는 방법은 아래와 같이 상하좌우(Up, Down, Left, Right)와 회전(Turn)이 있다.

 

예를 들면, 다음과 같다. (초기상태로부터의 이동)

이와 같은 방식으로 이동시킬 때에 그 움직일 위치에 다른 나무, 즉 1이 없어야만 움직일 수 있다. 그리고 움직임은 위의 그림과 같이 한 번에 한 칸씩만 움직인다. 단 움직이는 통나무는 어떤 경우이든지 중간단계에서 한 행이나 한 열에만 놓일 수 있다. 예를 들면 아래 그림에서 a와 같은 단계는 불가능하다. 그리고 회전의 경우에서는 반드시 중심점을 중심으로 90도 회전을 해야 한다. (항상 통나무의 길이가 3이므로 중심점이 있음)

그리고 이런 회전(Turn)이 가능하기 위해서는 그 통나무를 둘러싸고 있는 3*3 정사각형의 구역에 단 한 그루의 나무도 없어야만 한다. 즉, 아래그림 b, d와 같이 ?로 표시된 지역에 다른 나무, 즉 1이 없어야만 회전시킬 수 있다. 따라서 c와 같은 경우에, 통나무는 왼쪽 아직 벌채되지 않은 나무 때문에 회전시킬 수 없다.

문제는 통나무를 5개의 기본동작(U, D, L, R, T)만을 사용하여 처음위치(BBB)에서 최종위치(EEE)로 옮기는 프로그램을 작성하는 것이다. 단, 최소 횟수의 단위 동작을 사용해야 한다.
 

 

입력 및 출력

 

더보기

입력

 

첫째 줄에 주어진 평지의 한 변의 길이 N이 주어진다. (4 ≤ N ≤ 50) 주어진다. 이어서 그 지형의 정보가 0, 1, B, E로 이루어진 문자열로 주어진다. 한 줄에 입력되는 문자열의 길이는 N이며 입력 문자 사이에는 빈칸이 없다. 통나무와 최종 위치의 개수는 1개이다.

 

출력

 

첫째 줄에 최소 동작 횟수를 출력한다. 이동이 불가능하면 0만을 출력한다.

 

입력 예시

 

5
B0011
B0000
B0000
11000
EEE00

 

출력 예시

 

9

 

 


 

풀이

 

경로 탐색 문제 +a. 이왜골2 느낌을 받았다. 기본적인 2차원 경로 탐색을 수행하되, 다음과 같은 쟁점을 가진다.

 

  • 길이가 3이므로, 세로인 경우와 가로인 경우 2가지를 구분하여야 한다. 이동 시에도 중심축을 기준으로 이동하되 가로 및 세로 길이까지 고려하여 이동 가능 여부를 판별하여야 한다.
  • 회전의 경우 주변 9칸을 체크할 수 있어야 한다.
  • 좌표를 통해 초깃값 및 방향을 체크해야 한다. 내 풀이의 경우 무조건 평균값이 중심점이 됨을 이용하였다.

 

 

풀이 코드

from collections import deque
import sys 
input = sys.stdin.readline

dx = [0, 0, -1, 1]
dy = [-1, 1, 0, 0]
drx = [[-1, 0, 1], [0, 0, 0]]
dry = [[0, 0, 0], [-1, 0, 1]]

N = int(input())
map_list, B_list, E_list = list(), list(), list()

for i in range(N) :
  _map_list = input().strip()
  for j in range(N) :
    if _map_list[j] == 'E' :
      E_list.append((j, i))
    if _map_list[j] == 'B' :
      B_list.append((j, i))
  map_list.append(_map_list)

bx = (B_list[0][0] + B_list[1][0] + B_list[2][0]) // 3
by = (B_list[0][1] + B_list[1][1] + B_list[2][1]) // 3
br = 0 if B_list[0][0] != B_list[1][0] else 1

ex = (E_list[0][0] + E_list[1][0] + E_list[2][0]) // 3
ey = (E_list[0][1] + E_list[1][1] + E_list[2][1]) // 3
er = 0 if E_list[0][0] != E_list[1][0] else 1

q = deque([(bx, by, br, 0)])
visited = [[[False]*2 for _ in range(N)] for _ in range(N)]
visited[by][bx][br] = True

while q :
  x, y, r, dist = q.popleft()
  if (x, y, r) == (ex, ey, er) :
    print(dist)
    exit()
    
  for k in range(4) :
    ax, ay = x + dx[k], y + dy[k]
    flg = False
    for i in range(3) :
      arx, ary = ax + drx[r][i], ay + dry[r][i]
      if not (-1 < arx < N and -1 < ary < N and map_list[ary][arx] != '1') :
        flg = True
        break
    if flg or visited[ay][ax][r]:
      continue
    visited[ay][ax][r] = True
    q.append((ax, ay, r, dist+1))

  flg = False
  r = 1-r
  for ay in [y-1, y, y+1] :
    for ax in [x-1, x, x+1] :
      if not (-1 < ay < N and -1 < ax < N and map_list[ay][ax] != '1') :
        flg = True
        break
    if flg :
      break
  if flg or visited[y][x][r]:
    continue
  visited[y][x][r] = True
  q.append((x, y, r, dist+1))
print(0)

풀이 완료!

 

Contents

포스팅 주소를 복사했습니다

이 글이 도움이 되었다면 공감 부탁드립니다.