[백준] 12100번: 2048 (Easy)- 파이썬 (+ 2차원 배열 최대값 구하기)

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

12100번: 2048 (Easy)

문제

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

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

 

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

 

입력

첫째 줄에 보드의 크기 N (1 ≤ N ≤ 20)이 주어진다. 둘째 줄부터 N개의 줄에는 게임판의 초기 상태가 주어진다. 0은 빈 칸을 나타내며, 이외의 값은 모두 블록을 나타낸다. 블록에 쓰여 있는 수는 2보다 크거나 같고, 1024보다 작거나 같은 2의 제곱꼴이다. 블록은 적어도 하나 주어진다.

 

출력

최대 5번 이동시켜서 얻을 수 있는 가장 큰 블록을 출력한다.

 

풀이

살짝 복잡한 구현문제로 문제를 해결할 수 있는 방식을 생각해보고, 과정을 세분화해서 작성해나갔다. 

 

우선 왼쪽으로 1차원 배열을 이동시키는 방법을 생각하며 코드를 작성했다.

처음 작성한 코드는 stack을 사용해서 stack 가장 위에 있는 값과 같으면 그 값을 빼고 더해서 다시 넣는 방식으로 작성했으나, 

2 2 4

0 0 0

0 0 0

경우를 생각하지 못했다!

다시 작성한 코드로 통과를 했지만, 마음에 들지 않아서 더 깔끔하게 작성해봤다. 😁

 

답안 코드

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

N = int(input())
BOARD = [list(map(int, input().split())) for _ in range(N)]

def push(numbers):
    new_numbers = []
    idx = 0
    while idx < len(numbers):
        if idx < len(numbers)-1 and numbers[idx] == numbers[idx+1]:
            new_numbers.append(numbers[idx]*2)
            idx+=2
        else:
            new_numbers.append(numbers[idx])
            idx+=1
    new_numbers += [0] * (N-len(new_numbers))
    return new_numbers

def move(board, direction):
    new_board = [[]*N for _ in range(N)]

    if direction == 1:  # left
        for row in range(N):
            line = [board[row][col] for col in range(N) if board[row][col]]
            new_board[row] = push(line)
           
    elif direction == 2:    # right
        for row in range(N):
            line = [board[row][col] for col in range(N-1, -1, -1) if board[row][col]]
            new_board[row] = list(reversed(push(line)))
            
    elif direction == 3:    # up
        for col in range(N):
            line = [board[row][col] for row in range(N) if board[row][col]]
            new_line = push(line)
            for row in range(N):
                new_board[row].append(new_line[row])           

    else:
        for col in range(N):    # down
            line = [board[row][col] for row in range(N-1, -1, -1) if board[row][col]]
            new_line = list(reversed(push(line)))
            for row in range(N-1, -1, -1):
                new_board[row].append(new_line[row])

    return new_board

answer = 0
queue = deque()
queue.append((BOARD, 0))

while queue:
    board, depth = queue.popleft()

    if depth == 5:
        continue

    for direction in range(1, 5):   # 1 = left, 2 = right, 3 = up, 4 = down
        new_board = move(board, direction)
        queue.append((new_board, depth+1))
        answer = max(answer, max([max(row) for row in new_board]))

print(answer)

 

+) 2차원 배열에서 최대값을 구할 때 (feat.HW님)

max(max(2d_array))를 사용하면 안된다!

max([max(row) for row in 2d_array])를 사용할 것!

 

++) 더 쉽게 구하는 2차원 배열 최대값 (feat. JBJ님)

max(map(max, 2d_array))를 사용할 수 있다!