몽상실현개발주의

[BOJ] 2805 / 나무 자르기 / Python 파이썬 본문

Algorithm PS/BOJ

[BOJ] 2805 / 나무 자르기 / Python 파이썬

migrationArc 2021. 6. 18. 17:37

[BOJ] 2805 / 나무 자르기 / Python 파이썬

[BOJ] 2805 / 나무 자르기 / Python 파이썬

https://www.acmicpc.net/problem/2805

 

2805번: 나무 자르기

첫째 줄에 나무의 수 N과 상근이가 집으로 가져가려고 하는 나무의 길이 M이 주어진다. (1 ≤ N ≤ 1,000,000, 1 ≤ M ≤ 2,000,000,000) 둘째 줄에는 나무의 높이가 주어진다. 나무의 높이의 합은 항상 M보

www.acmicpc.net

 

풀이

이진탐색의 기본 문제이다.

 

주어진 나무의 높이들을 빼주면서 결과를 비교하여, 탐색 범위를 좁혀 주는 방식으로 이진탐색을 하였다.

 

  • res < M : 자를 높이 감소
  • res >= M : 자를 높이 증가

 

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

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

end = max(trees)
start = 1


def cutTree(h):
    global M
    res = 0
    for t in trees:
        if t-h > 0:
            res += (t-h)
    return res


while start <= end:
    mid = (start+end)//2

    res = cutTree(mid)

    if res < M:
        end = mid-1
    else:
        start = mid+1

print(end)
Comments