코딩테스트 대비/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)