BOJ-12100 2048(Easy) Swift

문제

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

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

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

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

 

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

입력

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

출력

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

내가 푼 풀이

- 다른 알고리즘 풀이방법이 떠오르지않아 위 아래 왼쪽 오른쪽 방향으로 이동하는 동작을 구현해서 모든 경우의수를 탐색했다.

- 최대 5번 움직일 수 있으니, 동작을 구현해서 5번 굴려줬다.

 

import Foundation

// 입력받기
let N = Int(readLine()!)!
var arr = [[Int]]()
var result = Int.min
for i in 0..<N {
    arr.append(readLine()!.split(separator: " ").map{Int(String($0))!})
}

// 움직이기
func move(count: Int, arr: [[Int]]) {
    // 5번 움직였다면 최댓값 구하고 끝내기
    if count == 5 {
        for i in 0..<N {
            result = max(result, arr[i].max()!)
        }
        return
    }
    
    //UP 위로이동
    var board = arr
    for i in 0..<N {
        var insert = [Int]()
        for j in 0..<N {
            // 0이 아닌값들을 저장
            if board[j][i] != 0 {
                insert.append(board[j][i])
                board[j][i] = 0
            }
        }
        // 모든값이 0이라면 insert에 저장되지않았으므로 패스
        if !insert.isEmpty {
            var index = 0
            var arrIndex = 0
            // 같은값이면 더해서 가장 위로 올려주기
            while arrIndex < insert.count - 1 {
                if insert[arrIndex] == insert[arrIndex+1] {
                    board[index][i] = insert[arrIndex] * 2
                    arrIndex += 2
                } else {
                    board[index][i] = insert[arrIndex]
                    arrIndex += 1
                }
                index += 1
            }
            // 맨 마지막번째 수가 합쳐지지않았다면 마지막번에 저장
            if arrIndex < insert.count {
                board[index][i] = insert[arrIndex]
            }
        }
    }
    // 위로 이동 후 이동한 보드를 그대로 다른방향으로 이동
    move(count: count+1, arr: board)
    
    //DOWN 아래로 이동
    board = arr
    for i in 0..<N {
        var insert = [Int]()
        for j in (0..<N).reversed() {
            if board[j][i] != 0 {
                insert.append(board[j][i])
                board[j][i] = 0
            }
        }
        if !insert.isEmpty {
            var index = N-1
            var arrIndex = 0
            while arrIndex < insert.count - 1 {
                if insert[arrIndex] == insert[arrIndex+1] {
                    board[index][i] = insert[arrIndex] * 2
                    arrIndex += 2
                } else {
                    board[index][i] = insert[arrIndex]
                    arrIndex += 1
                }
                index -= 1
            }
            if arrIndex < insert.count {
                board[index][i] = insert[arrIndex]
            }
        }
    }
    // 아래로 이동후 다른방향으로 이동
    move(count: count+1, arr: board)
    
    //LEFT 왼쪽 이동
    board = arr
    for i in 0..<N {
        var insert = [Int]()
        for j in 0..<N {
            if board[i][j] != 0 {
                insert.append(board[i][j])
                board[i][j] = 0
            }
        }
        if !insert.isEmpty {
            var index = 0
            var arrIndex = 0
            while arrIndex < insert.count - 1 {
                if insert[arrIndex] == insert[arrIndex+1] {
                    board[i][index] = insert[arrIndex] * 2
                    arrIndex += 2
                } else {
                    board[i][index] = insert[arrIndex]
                    arrIndex += 1
                }
                index += 1
            }
            if arrIndex < insert.count {
                board[i][index] = insert[arrIndex]
            }
        }
    }
    // 왼쪽으로 이동후 다른방향으로 이동
    move(count: count+1, arr: board)
    
    //RIGHT 오른쪽 이동
    board = arr
    for i in 0..<N {
        var insert = [Int]()
        for j in (0..<N).reversed() {
            if board[i][j] != 0 {
                insert.append(board[i][j])
                board[i][j] = 0
            }
        }
        if !insert.isEmpty {
            var index = N-1
            var arrIndex = 0
            while arrIndex < insert.count - 1 {
                if insert[arrIndex] == insert[arrIndex+1] {
                    board[i][index] = insert[arrIndex] * 2
                    arrIndex += 2
                } else {
                    board[i][index] = insert[arrIndex]
                    arrIndex += 1
                }
                index -= 1
            }
            if arrIndex < insert.count {
                board[i][index] = insert[arrIndex]
            }
        }
    }
    // 오른쪽으로 이동 후 다른방향으로 이동
    move(count: count+1, arr: board)
}

move(count: 0, arr: arr)
print(result)