[BOJ][S2] 마인크래프트 - 18111 (Kotlin)

[BOJ][S2] 마인크래프트 - 18111 (Kotlin)

문제 링크

18111번: 마인크래프트

분류

브루트포스 알고리즘, 구현

문제 설명

팀 레드시프트는 대회 준비를 하다가 지루해져서 샌드박스 게임인 ‘마인크래프트’를 켰다. 마인크래프트는 1 × 1 × 1(세로, 가로, 높이) 크기의 블록들로 이루어진 3차원 세계에서 자유롭게 땅을 파거나 집을 지을 수 있는 게임이다.

목재를 충분히 모은 lvalue는 집을 짓기로 하였다. 하지만 고르지 않은 땅에는 집을 지을 수 없기 때문에 땅의 높이를 모두 동일하게 만드는 ‘땅 고르기’ 작업을 해야 한다.

lvalue는 세로 N, 가로 M 크기의 집터를 골랐다. 집터 맨 왼쪽 위의 좌표는 (0, 0)이다. 우리의 목적은 이 집터 내의 땅의 높이를 일정하게 바꾸는 것이다. 우리는 다음과 같은 두 종류의 작업을 할 수 있다.

1. 좌표 (i, j)의 가장 위에 있는 블록을 제거하여 인벤토리에 넣는다.
2. 인벤토리에서 블록 하나를 꺼내어 좌표 (i, j)의 가장 위에 있는 블록 위에 놓는다.

1번 작업은 2초가 걸리며, 2번 작업은 1초가 걸린다. 밤에는 무서운 몬스터들이 나오기 때문에 최대한 빨리 땅 고르기 작업을 마쳐야 한다. ‘땅 고르기’ 작업에 걸리는 최소 시간과 그 경우 땅의 높이를 출력하시오.

단, 집터 아래에 동굴 등 빈 공간은 존재하지 않으며, 집터 바깥에서 블록을 가져올 수 없다. 또한, 작업을 시작할 때 인벤토리에는 B개의 블록이 들어 있다. 땅의 높이는 256블록을 초과할 수 없으며, 음수가 될 수 없다.

입력

첫째 줄에 N, M, B가 주어진다. (1 ≤ M, N ≤ 500, 0 ≤ B ≤ 6.4 × 107)

둘째 줄부터 N개의 줄에 각각 M개의 정수로 땅의 높이가 주어진다. (i + 2)번째 줄의 (j + 1)번째 수는 좌표 (i, j)에서의 땅의 높이를 나타낸다. 땅의 높이는 256보다 작거나 같은 자연수 또는 0이다.

출력

첫째 줄에 땅을 고르는 데 걸리는 시간과 땅의 높이를 출력하시오. 답이 여러 개 있다면 그중에서 땅의 높이가 가장 높은 것을 출력하시오.

풀이

코틀린으로 PS를 연습해볼려고 만만한 실버문제를 아무거나 골랐다. 근데 생각보다 풀이가 생각이 잘 나지 않아서 당황한 문제.
최대 높이가 256 정도 밖에 안되고 배열의 크기도 500 x 500 이므로 그냥 깡 구현하면 되는 문제이다.

C++로 할때랑 다르게 입력받을 수 있는 방법이 다양하다보니 여러 사람의 코드를 참조했다.
많은 사람들이 BufferedReader(InputStreamReader(System.in)의 readLine()!!으로 string을 입력받고 split() 또는 StringTokenizer을 사용하여 나누어 사용하는 듯 한데, 공식 문서를 찾아보니 코틀린 1.6 이후로는 그냥 readln() 쓰면 되는 것 같다. (왜 다들 힘들게 긴 줄 다 쓰고 있는지 잘 모르겠다.)

성능 상 split은 regex를 사용하기에 단일 seperator와 같은 경우는 StringTokenizer가 성능이 더 좋다고는 하는데 직접 문제 풀면서 돌려보니까 큰 차이는 나지 않는 듯 하다. 일단은 그냥 split으로 사용하려고 한다.

    var candidate = Pair(Int.MAX_VALUE, 0)

    for(height in minHeight..maxHeight) {
        var stackCount = 0
        var popCount = 0

        for(row in arr) {
            for(target in row) {
                when {
                    target < height -> stackCount += height - target
                    target > height -> popCount += target - height
                }
            }
        }

        val time = stackCount + popCount * 2

        if(b + popCount >= stackCount && time <= candidate.first)
            candidate = Pair(time, height)
    }

핵심 로직은 위 코드에 다 담겨 있다.
가장 낮은 높이부터 가장 높은 위치의 블록까지 모두 평탄화 하면서 가장 시간이 적게 들고, 높이가 높은 경우를 구한다.

다른 사람들의 코드를 살펴보니 블록 합을 맵 넓이로 나눠 평균 높이를 구해 maxHeight로 시간을 단축한 사람들이 있어 따라해봤다.

코드

풀고 나니까 드는 생각인데 어짜피 특정 인덱스에 접근하지 않고 순회만 하기때문에 Array말고 그냥 List에 담아도 될거 같다.