티스토리 뷰

728x90

문제

www.acmicpc.net/problem/2805

 

2805번: 나무 자르기

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

www.acmicpc.net

 

소스코드

n, m = map(int, input().split())
tree = list(map(int, input().split()))

start = 1n, m = map(int, input().split())
tree = list(map(int, input().split()))

start = 1
end = max(tree)

while start<=end:
    mid = (start+end)//2
    sum = 0
    for i in tree:
        if i>mid:
            sum += i-mid
    if sum>=m:
        start = mid+1
    else:
        end = mid-1

print(end)
end = max(tree)

while start <= end:
    mid = (start+end)//2
    sum = 0
    for i in tree:
        if i-mid > 0:
            sum += i-mid
    
    if sum == m:
        break
    elif sum > m:
        start = mid+1
    else:
        end = mid-1

print(mid)
반응형