[BOJ] 2805번: 나무 자르기

✏️ 문제

상근이는 나무 M미터가 필요하다. 근처에 나무를 구입할 곳이 모두 망해버렸기 때문에, 정부에 벌목 허가를 요청했다. 정부는 상근이네 집 근처의 나무 한 줄에 대한 벌목 허가를 내주었고, 상근이는 새로 구입한 목재절단기를 이용해서 나무를 구할 것이다.

목재절단기는 다음과 같이 동작한다. 먼저, 상근이는 절단기에 높이 H를 지정해야 한다. 높이를 지정하면 톱날이 땅으로부터 H미터 위로 올라간다. 그 다음, 한 줄에 연속해있는 나무를 모두 절단해버린다. 따라서, 높이가 H보다 큰 나무는 H 위의 부분이 잘릴 것이고, 낮은 나무는 잘리지 않을 것이다. 예를 들어, 한 줄에 연속해있는 나무의 높이가 20, 15, 10, 17이라고 하자. 상근이가 높이를 15로 지정했다면, 나무를 자른 뒤의 높이는 15, 15, 10, 15가 될 것이고, 상근이는 길이가 5인 나무와 2인 나무를 들고 집에 갈 것이다. (총 7미터를 집에 들고 간다) 절단기에 설정할 수 있는 높이는 양의 정수 또는 0이다.

상근이는 환경에 매우 관심이 많기 때문에, 나무를 필요한 만큼만 집으로 가져가려고 한다. 이때, 적어도 M미터의 나무를 집에 가져가기 위해서 절단기에 설정할 수 있는 높이의 최댓값을 구하는 프로그램을 작성하시오.

 

🤖 입출력 예시

 

🧐 난이도/소요 시간

• 난이도: solved.ac 기준 S2
• 소요 시간: 19분 → 어제 랜선 자르기를 풀고 봐서 그렇지 이 문제를 먼저 봤다면 측정 불가였을 듯

 

💡 풀이

어제 올린 '랜선 자르기'와 동일한 매개변수 탐색 유형이다. 그래서 코드 틀은 그대로 쓰고 필요한 부분만 추가했다.

 

(1) 코드 by 나

import sys

input = sys.stdin.readline

N, M = map(int, input().split())

tree_heights = list(map(int, input().split()))

start, end = 1, max(tree_heights)

while start <= end:
    mid = (start+end)//2
    
    machine_height = 0
    
    for tree in tree_heights:
        if tree >= mid:
            machine_height += (tree-mid)
    
    if machine_height >= M:
        start = mid+1
    else:
        end = mid-1

print(end)

 

Line 17: 잘리고 남은 나무의 길이가 0 미만일 수 없기 때문에 나무 높이가 mid보다 큰 경우만 처리해서 상근이가 가져가는 나무에 더해줬다. 이 부분이 없으니까 예제 답이 이상하게 나왔음

 

근데 이 코드로 통과는 했는데 시간이 무려 4224ms나 걸려서 챗지피티가 알려준 코드를 넣어보니까 수행 시간이 절반 넘게 줄었다.

 

(2) 코드 by chatGPT

def get_wood_length(trees, height):
    total = 0
    for tree in trees:
        if tree > height:
            total += tree - height
    return total

def find_max_saw_height(trees, target):
    left, right = 0, max(trees)
    result = 0
    
    while left <= right:
        mid = (left + right) // 2
        wood = get_wood_length(trees, mid)
        
        if wood >= target:
            result = mid  # 현재 높이에서 나무를 충분히 가져갈 수 있으면, 더 높은 높이도 확인
            left = mid + 1
        else:
            right = mid - 1
    
    return result

# 입력 받기
N, M = map(int, input().split())
trees = list(map(int, input().split()))

# 최대 절단기 높이 출력
print(find_max_saw_height(trees, M))

 

내가 생각하는 이 코드와 내 코드의 가장 큰 차이점은 나무마다 잘라내는 과정의 함수화이다. 바로 챗지피티한테 두 코드의 수행 시간이 이렇게나 차이가 많이 나는 이유 물어봄

 

 

프로그래머스에서 문제 풀 때는 좀 덜한데, 백준에서 문제 풀 때는 함수를 거의 사용하지 않는다. 함수를 잘 써야겠다고 다짐하게 되는 문제였다.