🏄♀️ 코딩테스트/🐍 Python
[ 백준 16197 해설 ] ( python ) 두 동전
Po_tta_tt0
2022. 8. 10. 18:38
반응형
📚 두 동전
문제
N×M 크기의 보드와 4개의 버튼으로 이루어진 게임이 있다. 보드는 1×1크기의 정사각형 칸으로 나누어져 있고, 각각의 칸은 비어있거나, 벽이다. 두 개의 빈 칸에는 동전이 하나씩 놓여져 있고, 두 동전의 위치는 다르다.
버튼은 "왼쪽", "오른쪽", "위", "아래"와 같이 4가지가 있다. 버튼을 누르면 두 동전이 버튼에 쓰여 있는 방향으로 동시에 이동하게 된다.
- 동전이 이동하려는 칸이 벽이면, 동전은 이동하지 않는다.
- 동전이 이동하려는 방향에 칸이 없으면 동전은 보드 바깥으로 떨어진다.
- 그 외의 경우에는 이동하려는 방향으로 한 칸 이동한다.이동하려는 칸에 동전이 있는 경우에도 한 칸 이동한다.
두 동전 중 하나만 보드에서 떨어뜨리기 위해 버튼을 최소 몇 번 눌러야하는지 구하는 프로그램을 작성하시오.
입력
첫째 줄에 보드의 세로 크기 N과 가로 크기 M이 주어진다. (1 ≤ N, M ≤ 20)
둘째 줄부터 N개의 줄에는 보드의 상태가 주어진다.
- o: 동전
- .: 빈 칸
- #: 벽
동전의 개수는 항상 2개이다.
출력
첫째 줄에 두 동전 중 하나만 보드에서 떨어뜨리기 위해 눌러야 하는 버튼의 최소 횟수를 출력한다. 만약, 두 동전을 떨어뜨릴 수 없거나, 버튼을 10번보다 많이 눌러야 한다면, -1을 출력한다.
예제 입력 1 복사
1 2
oo
예제 출력 1 복사
1
예제 입력 2 복사
6 2
.#
.#
.#
o#
o#
##
예제 출력 2 복사
4
예제 입력 3 복사
6 2
..
..
..
o#
o#
##
예제 출력 3 복사
3
예제 입력 4 복사
5 3
###
.o.
###
.o.
###
예제 출력 4 복사
-1
예제 입력 5 복사
5 3
###
.o.
#.#
.o.
###
예제 출력 5 복사
3
✍ 접근
- 재귀를 돌리면서 조건을 만족시켰을 때의 값을 res에 넣자.
- 재귀를 돌리면서 조건을 만족하지 못할 것들은 빨리 return시키자
정답코드
import sys
sys.setrecursionlimit(10**6)
input = sys.stdin.readline
N,M = map(int,input().split())
board = [list(input().rstrip()) for _ in range(N)]
visited = []
coinpos = []
dx = [-1,0,1,0]
dy = [0,-1,0,1]
res = 2147000000
def DFS(coin1, coin2, cnt):
global res
x1,y1 = coin1
x2,y2 = coin2
if cnt>=res or cnt>10:
return
if x1 == x2 and y1 == y2:
return
if (0 > x1 or N <= x1 or 0 > y1 or M <= y1) and (0 > x2 or N <= x2 or 0 > y2 or M <= y2):
return
if (0 > x1 or N <= x1 or 0 > y1 or M <= y1) or (0 > x2 or N <= x2 or 0 > y2 or M <= y2):
res = min(res,cnt)
return
for i in range(4):
nx1 = x1 + dx[i]
ny1 = y1 + dy[i]
nx2 = x2 + dx[i]
ny2 = y2 + dy[i]
if 0<=nx1<N and 0<=ny1<M and 0<=nx2<N and 0<=ny2<M:
if board[nx1][ny1] != '#' and board[nx2][ny2] != '#' and (nx1,ny1,nx2,ny2) not in visited:
visited.append((nx1,ny1,nx2,ny2))
DFS((nx1,ny1),(nx2,ny2), cnt+1)
visited.pop()
elif board[nx1][ny1] == '#' and board[nx2][ny2] != '#' and (x1,y1,nx2,ny2) not in visited:
visited.append((x1,y1,nx2,ny2))
DFS((x1,y1),(nx2,ny2),cnt+1)
visited.pop()
elif board[nx1][ny1] != '#' and board[nx2][ny2] == '#' and (nx1,ny1,x2,y2) not in visited:
visited.append((nx1,ny1,x2,y2))
DFS((nx1,ny1),(x2,y2),cnt+1)
visited.pop()
else:
visited.append((nx1,ny1,nx2,ny2))
DFS((nx1,ny1),(nx2,ny2), cnt+1)
visited.pop()
for i in range(N):
for j in range(M):
if board[i][j] == 'o':
coinpos.append((i,j))
DFS(coinpos[0],coinpos[1],0)
print(res) if res<=10 else print(-1)
해설
아주 길죠?
DFS가 아니라 BFS로도 가능할 것 같습니다.
DFS에서의 예외처리가 중요한 문제였습니다
⭐ 배움
- DFS의 예외처리는 중요하다!
반응형