백준 12100 파이썬 - 2048 (Easy)

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

12100번: 2048 (Easy)

문제

2048 게임은 4×4 크기의 보드에서 혼자 즐기는 재미있는 게임이다. 이 링크를 누르면 게임을 해볼 수 있다.

이 게임에서 한 번의 이동은 보드 위에 있는 전체 블록을 상하좌우 네 방향 중 하나로 이동시키는 것이다. 이때, 같은 값을 갖는 두 블록이 충돌하면 두 블록은 하나로 합쳐지게 된다. 한 번의 이동에서 이미 합쳐진 블록은 또 다른 블록과 다시 합쳐질 수 없다. (실제 게임에서는 이동을 한 번 할 때마다 블록이 추가되지만, 이 문제에서 블록이 추가되는 경우는 없다)

<그림 1>의 경우에서 위로 블록을 이동시키면 <그림 2>의 상태가 된다. 여기서, 왼쪽으로 블록을 이동시키면 <그림 3>의 상태가 된다.

<그림 4>의 상태에서 블록을 오른쪽으로 이동시키면 <그림 5>가 되고, 여기서 다시 위로 블록을 이동시키면 <그림 6>이 된다. 여기서 오른쪽으로 블록을 이동시켜 <그림 7>을 만들 수 있다.

<그림 8>의 상태에서 왼쪽으로 블록을 옮기면 어떻게 될까? 2가 충돌하기 때문에, 4로 합쳐지게 되고 <그림 9>의 상태가 된다.

<그림 10>에서 위로 블록을 이동시키면 <그림 11>의 상태가 된다. 

<그림 12>의 경우에 위로 블록을 이동시키면 <그림 13>의 상태가 되는데, 그 이유는 한 번의 이동에서 이미 합쳐진 블록은 또 합쳐질 수 없기 때문이다.

마지막으로, 똑같은 수가 세 개가 있는 경우에는 이동하려고 하는 쪽의 칸이 먼저 합쳐진다. 예를 들어, 위로 이동시키는 경우에는 위쪽에 있는 블록이 먼저 합쳐지게 된다. <그림 14>의 경우에 위로 이동하면 <그림 15>를 만든다.

이 문제에서 다루는 2048 게임은 보드의 크기가 N×N 이다. 보드의 크기와 보드판의 블록 상태가 주어졌을 때, 최대 5번 이동해서 만들 수 있는 가장 큰 블록의 값을 구하는 프로그램을 작성하시오.

코드

import sys
from copy import deepcopy
n = int(sys.stdin.readline())
board = [list(map(int, sys.stdin.readline().split())) for _ in range(n)]
answer = 0
def move(dir):
    if dir == 0:
        for i in range(n):
            idx = 0
            for j in range(1, n):
                if board[j][i]:
                    tmp = board[j][i]
                    board[j][i] = 0
                    if board[idx][i] == 0:
                        board[idx][i] = tmp
                    elif board[idx][i] == tmp:
                        board[idx][i] = tmp * 2
                        idx += 1
                    else:
                        idx += 1
                        board[idx][i] = tmp
    elif dir == 1:
        for i in range(n):
            idx = n - 1
            for j in range(n - 2, -1, -1):
                if board[j][i]:
                    tmp = board[j][i]
                    board[j][i] = 0
                    if board[idx][i] == 0:
                        board[idx][i] = tmp
                    elif board[idx][i] == tmp:
                        board[idx][i] = tmp * 2
                        idx -= 1
                    else:
                        idx -= 1
                        board[idx][i] = tmp
    elif dir == 2:
        for j in range(n):
            idx = 0
            for i in range(1, n):
                if board[j][i]:
                    tmp = board[j][i]
                    board[j][i] = 0
                    if board[j][idx] == 0:
                        board[j][idx] = tmp
                    elif board[j][idx] == tmp:
                        board[j][idx] = tmp * 2
                        idx += 1
                    else:
                        idx += 1
                        board[j][idx] = tmp
    else:
        for j in range(n):
            idx = n - 1
            for i in range(n - 2, -1, -1):
                if board[j][i]:
                    tmp = board[j][i]
                    board[j][i] = 0
                    if board[j][idx] == 0:
                        board[j][idx] = tmp
                    elif board[j][idx] == tmp:
                        board[j][idx] = tmp * 2
                        idx -= 1
                    else:
                        idx -= 1
                        board[j][idx] = tmp
def dfs(count):
    global board, answer
    if count == 5:
        for i in range(n):
            boardMax = max(board[i])
            answer = max(answer, boardMax)
        return
    temp = deepcopy(board)
    for i in range(4):
        move(i)
        dfs(count + 1)
        board = deepcopy(temp)
dfs(0)
print(answer)

풀이

move함수에서는 dir이 가능한 4가지경우(네 방향)에 따라 움직이는 연산과 합쳐지는 연산을 수행한다. dfs함수에서 네 방향으로 움직여보면서 움직일 때 마다 count를 1씩 증가시키고 5가 되면 5번 움직였다는 것이므로 최댓값을 구한다. 그 후 최댓값을 출력하면 된다.