코딩테스트 대비/BOJ

[Baekjoon/Python] 2805번: 나무 자르기 - 효과는 굉장했다!

bluetag_boy 2021. 11. 2. 23:12
반응형
 

2805번: 나무 자르기

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

www.acmicpc.net

알고리즘 분류

  • 이분 탐색

 

SOLUTION

# 시간초과로 인해 pypy3으로 제출

import sys

N, M = map(int, sys.stdin.readline().split())
tree = list(map(int, sys.stdin.readline().split()))

low, high = 1 , max(tree) 

while low <= high: # 이분 탐색
    mid = (low + high) // 2 

    total = 0
    for i in tree:
        if i >= mid:
            total += i - mid # 나무를 자르고 가져갈 수 있는 총 길이
            # print(total, mid, high)
    
    if total >= M: # 가져갈 수 있는 총 나무 길이가 M보다 같거나 많을 시
        low = mid + 1 # 더 길게 나무를 잘라야 하므로 low = mid + 1 

    else: # 가져갈 수 있는 총 나무 길이가 M보다 작다면
        high = mid -1 # 더 짧게 나무를 잘라야 하므로 high = mid -1

print(high)