[BOJ][S2] 마인크래프트 - 18111 (Kotlin)
[BOJ][S2] 마인크래프트 - 18111 (Kotlin)
분류
브루트포스 알고리즘, 구현
문제 설명
팀 레드시프트는 대회 준비를 하다가 지루해져서 샌드박스 게임인 ‘마인크래프트’를 켰다. 마인크래프트는 1 × 1 × 1(세로, 가로, 높이) 크기의 블록들로 이루어진 3차원 세계에서 자유롭게 땅을 파거나 집을 지을 수 있는 게임이다.
목재를 충분히 모은 lvalue는 집을 짓기로 하였다. 하지만 고르지 않은 땅에는 집을 지을 수 없기 때문에 땅의 높이를 모두 동일하게 만드는 ‘땅 고르기’ 작업을 해야 한다.
lvalue는 세로 N, 가로 M 크기의 집터를 골랐다. 집터 맨 왼쪽 위의 좌표는 (0, 0)이다. 우리의 목적은 이 집터 내의 땅의 높이를 일정하게 바꾸는 것이다. 우리는 다음과 같은 두 종류의 작업을 할 수 있다.
- 좌표 (i, j)의 가장 위에 있는 블록을 제거하여 인벤토리에 넣는다.
- 인벤토리에서 블록 하나를 꺼내어 좌표 (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에 담아도 될거 같다.