파이썬으로 풀어보는 백준 17837번: 새로운 게임 2 (삼성 A형 기출 문제)

https://www.acmicpc.net/problem/17837

17837번: 새로운 게임 2

 

내 풀이:

def white(y, x, ny, nx, idx):
    a = 999999
    lst = []
    while a != idx:
        a = arr[y][x].pop()
        lst.insert(0, a)
 
    for num in lst:
        units[num][0] = ny
        units[num][1] = nx
    arr[ny][nx].extend(lst)
    if len(arr[ny][nx]) >= 4:
        print(ans)
        exit()
 
 
def red(y, x, ny, nx, idx):
    a = 999999
    lst = []
    while a != idx:
        a = arr[y][x].pop()
        lst.append(a)
    for num in lst:
        units[num][0] = ny
        units[num][1] = nx
    arr[ny][nx].extend(lst)
    if len(arr[ny][nx]) >= 4:
        print(ans)
        exit()
 
 
def blue(idx):
    if units[idx][2] == 0:
        units[idx][2] = 1
    elif units[idx][2] == 1:
        units[idx][2] = 0
    elif units[idx][2] == 2:
        units[idx][2] = 3
    elif units[idx][2] == 3:
        units[idx][2] = 2
 
 
def move():
    global ans
    ans += 1
    for idx, u in enumerate(units):
        y = u[0]
        x = u[1]
        dir = u[2]
 
        ny = y + dy[dir]
        nx = x + dx[dir]
 
        if 0 <= ny <= N-1 and 0 <= nx <= N-1:
            if color[ny][nx] == 0:
                white(y, x, ny, nx, idx)
 
            elif color[ny][nx] == 1:
                red(y, x, ny, nx, idx)
 
            elif color[ny][nx] == 2:
 
                ny = y - dy[dir]
                nx = x - dx[dir]
                blue(idx)
                if not 0 <= ny <= N-1 or not 0 <= nx <= N-1 or color[ny][nx] == 2:
                    pass
 
                elif 0 <= ny <= N-1 and 0 <= nx <= N-1:
                    if color[ny][nx] == 0:
                        white(y, x, ny, nx, idx)
 
                    elif color[ny][nx] == 1:
                        red(y, x, ny, nx, idx)
 
        else:
            ny = y - dy[dir]
            nx = x - dx[dir]
            blue(idx)
            if not 0 <= ny <= N - 1 or not 0 <= nx <= N - 1 or color[ny][nx] == 2:
                pass
 
            elif 0 <= ny <= N - 1 and 0 <= nx <= N - 1:
                if color[ny][nx] == 0:
                    white(y, x, ny, nx, idx)
 
                elif color[ny][nx] == 1:
                    red(y, x, ny, nx, idx)
 
    if ans == 996:
        print(-1)
        exit()
    move()
 
 
def minus1(num):
    return num-1
 
 
N, K = map(int,input().split())
color = [list(map(int,input().split())) for _ in range(N)]
arr = [[[] for __ in range(N)] for _ in range(N)]
units = [list(map(minus1,map(int,input().split()))) for _ in range(K)]
 
for idx, i in enumerate(units):
    arr[i[0]][i[1]].append(idx)
 
dy = [0, 0, -1, 1]
dx = [1, -1, 0, 0]
ans = 0
 
move()

Python3, 76ms

게임 규칙이 다소 복잡하여 구현하기까지 디버깅을 많이 필요로 하였다. 사실 말을 쌓고 옮기는 과정에서 수많은 pop과 append, insert를 이용해야 했기 때문에 다른 방법이 있지는 않을까 고민하는데 시간을 더 오래 사용한 문제가 되겠다. 결국 다른 방법이 없다고 판단하여 게임 규칙을 색깔별로 함수로 만들어 구현하였다. 문제는 원래 재귀 깊이에 있었는데 문제 조건에서는 답이 1000 이하인 경우에는 답을 출력하고 1000을 넘는 경우에만 -1을 출력하도록 했는데 위의 코드는 답이 997 이상인 경우에는 -1을 출력하도록 하였다. 그런데도 정답처리를 받은 것으로 보아 샘플에는 답이 997 ~ 1000 인 데이터는 없었던 것으로 판단된다. 만약 정확하게 문제 조건을 구현하려면 색깔별 함수를 제거하고 직접 코드로 구현하거나 아니면 move()를 재귀 형태가 아닌 반복문의 형태로 구현해야 할 것 같다.

 

덧붙이자면 이번에 코딩을 하면서 위에 코드에서 보이는 exit() 와 이전에 시도한 try, except가 있었는데 exit()를 사용하면 프로그램이 종료되는 것이 아니라 except로 가는 것을 확인하였다. except는 에러가 뜰 경우에만 작동하는 줄 알았는데 프로그램을 강제로 나가게 할 때 작동하는 것으로 보인다.