[백준 python 18111] 마인크래프

[Silver II] 마인크래프트 - 18111

문제 링크

성능 요약

메모리: 118612 KB, 시간: 404 ms

분류

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

제출 일자

2024년 10월 18일 23:19:34

문제 설명

팀 레드시프트는 대회 준비를 하다가 지루해져서 샌드박스 게임인 ‘마인크래프트’를 켰다. 마인크래프트는 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 × 10⁷)

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

출력

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

풀이

import sys
input = sys.stdin.readline

n, m, b = map(int, input().split())
lands = []
for _ in range(n):
    lands += list(map(int, input().split()))
    
min_time = 500*500*2*257
max_height = 0
sum_land = sum(lands)

for h in range(max(lands), min(lands)-1, -1):
    if sum_land + b >= h * n * m:
        time = 0
        for land in lands:
            diff = land - h
            if diff > 0:
                time += diff * 2
            elif diff < 0:
                time -= diff
                
        if time < min_time:
            min_time = time
            max_height = h
            
print(min_time, max_height)

 

이건 풀다가 시간초과가 발생하고 계속 안풀려서 인터넷 풀이를 확인했다.

풀이에서는 2차원 배열을 1차원으로 변경하였다.

`min_time`은 땅을 고르는 데 걸리는 시간이고, `max_height`는 땅의 높이이다.

1부터 256까지 다 돌리는 것을 시간초과가 나기 떄문에 땅의 최댓값과 최소값을 잡고 반복문을 돌렸다.

첫 번째 if문에서의 조건인 `sum_land + b >= h * n * m`은 수식의 왼쪽은 입력값으로 주어진 땅들의 높이들과 인벤토리에 들어가 있는 블록의 개수를 더한 값, 즉 현재 소유하고 있는 블록의 개수를 말한다. 수식의 오른쪽은 평탄화된 땅에 사용된 블록의 개수이다. 소유하고 있는 블록의 개수가 평탄화된 땅의 블록의 개수보다 작으면 평탄화할 수 없기 때문에 이러한 조건식이 된다.

땅의 높이를 차례대로 가져와 높이를 뺀 값이 양수이면 시간이 2배걸리고 음수이면 그냥 1배이다.

이걸 다 돌고 최소 시간을 구해야 하기 때문에 미리 최댓값으로 초기화해 둔 min_time이 for문을 돌며 구한 time보다 크면 치환해준다.

 

풀이를 보고 내가 이해를 하고 다시 풀이를 해보니 내가 문제를 잘 이해하지 못했던 것 같다.

그리고 이 문제는 파이썬은 느리기 때문에 pypy를 사용해야 시간초과가 발생하지 않는다.