떡볶이 떡 만들기(파이썬) / 이진탐색

문제

• 오늘 동빈이는 여행 가신 부모님을 대신해서 떡집 일을 하기로 했다. 오늘은 떡볶이 떡을 만드는 날이다.
  동빈이네 떡볶이 떡은 재밌게도 떡볶이 떡의 길이가 일정하지 않다. 대신에 한봉지 안에 들어가는
  떡의 총 길이는 절단기로 잘라서 맞춰준다
• 절단기에 높이(H) 를 지정하면 줄지어진 떡을 한 번에 절단한다. 높이가 H보다 긴 떡은 H 위의 부분이
  잘릴 것이고, 낮은 떡은 잘리지 않는다
• 예를 들어 높이가 19, 14, 10, 17cm인 떡이 나란히 있고 절단기 높이를 15cm로 지정하면 자른 뒤
  떡의 높이는 15, 14, 10, 15cm가 될 것이다. 잘린 떡의 길이는 차례대로 4, 0, 0, 2cm이다.
  손님은 6cm만큼의 길이를 가져간다
• 손님이 왔을 때 요청한 총 길이가 M일 때 적어도 M만큼의 떡을 얻기 위해 절단기에 설정할 수 있는 높이의
  최댓값을 구하는 프로그램을 작성하세요

"INPUT"
4 6
19 15 10 17

"OUTPUT"
15

 

 

풀이

계수정렬을 통해 끝에서부터 잘라본다.

이 때, 원하는 길이보다 큰 값을 발견했을 때, 탐색을 중단하고 정답을 반환한다.

import sys

sys.stdin = open("input_6-3.txt","r")
sys.stdout = open("output_6-3.txt","w")

# input
n , m = map(int,input().split())
rc_list = list(map(int,input().split()))
answer = 0

# 계수 정렬
count_list = [0] * (max(rc_list)+1)
for rc in rc_list:
    count_list[rc] +=1

# 제시한 값보다 큰 값을 찾는 최대값(뒤에서부터)
for i in range(max(rc_list), -1, -1):
    # 값 계산
    sum = 0
    for j in range(max(rc_list), i, -1):
        sum += (count_list[j] * (j-i)) # 차액 더해주기

    # 기준이 되면
    if sum >= m:
        answer = i
        break

print(answer)