[파이썬 알고리즘 문제풀이] - Section4 / 이분탐색(결정알고리즘) & 그리디 알고리즘 - 3

Chooooo·2023년 1월 27일
0

🎈 뮤직비디오(결정알고리즘)

지니레코드에서는 불세출의 가수 조영필의 라이브 동영상을 DVD로 만들어 판매하려 한다.
DVD에는 총 N개의 곡이 들어가는데, DVD에 녹화할 때에는 라이브에서의 순서가 그대로 유지
되어야 한다. 순서가 바뀌는 것을 우리의 가수 조영필씨가 매우 싫어한다. 즉, 1번 노래와 5번
노래를 같은 DVD에 녹화하기 위해서는 1번과 5번 사이의 모든 노래도 같은 DVD에 녹화해야
한다. 또한 한 노래를 쪼개서 두 개의 DVD에 녹화하면 안된다.
지니레코드 입장에서는 이 DVD가 팔릴 것인지 확신할 수 없기 때문에 이 사업에 낭비되는
DVD를 가급적 줄이려고 한다. 고민 끝에 지니레코드는 M개의 DVD에 모든 동영상을 녹화하기
로 하였다. 이 때 DVD의 크기(녹화 가능한 길이)를 최소로 하려고 한다. 그리고 M개의 DVD는
모두 같은 크기여야 제조원가가 적게 들기 때문에 꼭 같은 크기로 해야 한다.

▣ 입력설명
첫째 줄에 자연수 N(1≤N≤1,000), M(1≤M≤N)이 주어진다. 다음 줄에는 조영필이 라이브에서
부른 순서대로 부른 곡의 길이가 분 단위로(자연수) 주어진다. 부른 곡의 길이는 10,000분을
넘지 않는다고 가정하자.

▣ 출력설명
첫 번째 줄부터 DVD의 최소 용량 크기를 출력하세요.

▣ 입력예제 1
9 3
1 2 3 4 5 6 7 8 9

▣ 출력예제 1
17

import sys
# sys.stdin = open("input.text", "rt")

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

start = 1
end = sum(data)  #모든 노래가 하나의 DVD에 들어가는게 그 DVD의 최대 길이

def count_max(len): #해당 디비디 길이 들어왔을 때 만들 수 있는 디비디 개수
    cnt = 1
    sum_data = 0
    for i in range(N):
        if sum_data + data[i] <= len: #같아도 저장되잖아! 예외...
            sum_data += data[i]
        else:
            cnt += 1
            sum_data = data[i]
    return cnt

res = 24242424242424
while start <= end:
    mid = (start + end) // 2  #최소용량

    if count_max(mid) <= M:  #M보다 작아도 저장되는건 M으로도 되는거잖아
        res = mid #용량의 최소를 찾는 것이기에 더 좋은 답이 없는지 유효성 검사 해야지
        end = mid -1
    else: #용량이 너무 작아서 M보다 더 필요한거야 답의 범위가 유효하지 않음
        start = mid + 1


print(res)
        

🎃 코멘트
DVD 용량이 최소가 되어야 한다. 답의 범위는 1~max(data)이다. 이유는 노래를 한 디비디에 다 담을 경우가 용량 최대가 되겠지.
이제 이진 탐색을 하면서 해당 중간값이 M개로 디비디를 다 담을 수 있을지 확인하는거야!

  • 만약 M개보다 적게 담을 수 있는 경우는 당연히 담을 수 있는거야

9곡을 저장하는데 2장에 저장할 수 있다는 것은 3장에는 당연히 저장할 수 있다고 생각해야한다.
예를 들어 5장이 필요하다면 3장으로는 당연히 안되는거잖아.(최소 5장이니까)

이진탐색은 기본적으로 자료가 정렬되어 있어야 가능한 알고리즘이다. 하지만 이런 문제는 입력되는 N개의 데이터를 이진 탐색하는게 아니다.

  • 답이 있을거라 확신하는 범위인 start~end까지의 자연수를 이진탐색하는 것이다!

어쨋든 특수한 경우를 이해해야 한다.

profile
back-end, 지속 성장 가능한 개발자를 향하여

0개의 댓글