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

 

알고리즘 입문 수업을 듣고 중요한 내용을 정리했습니다.
개인 공부 후 자료를 남기기 위한 목적이므로 내용 상에 오류가 있을 수 있습니다.

 

문제

[뮤직 비디오]

 

AAA레코드에서는 불세출의 가수 BBB의 라이브 동영상을 DVD로 마들어 판매하려고 한다.

DVD에는 총 N개의 곡이 들어가는데, DVD에 녹화할 때에는 라이브에서의 순서가 그대로 유지되어야 한다.

순서가 바뀌는 것을 가수 BBB가 매우 싫어한다.

즉, 1번 노래와 5번 노래를 같은 DVD에 녹화하기 위해서는 1번과 5번 사이의 모든 노래도 같은 DVD에 녹화해야 한다.

또한, 한 노래를 쪼개서 두 개의 DVD에 녹화하면 안된다.

 

AAA레코드 입장에서는 이 DVD가 팔릴 것인지 확신할 수 없기 때문에 이 사업에 낭비되는 DVD를 가급적 줄이려고 한다.

고민 끝에 AAA레코드는 M개의 DVD에 모든 동영상을 녹화하기로 하였다. 이 때 DVD의 크기(녹화 가능한 길이)를 최소로 하려고 한다.

그리고 M개의 DVD는 모두 같은 크기여야 제조원가가 적게 들기 때문에 꼭 같은 크기로 해야 한다.

 

 

*입력 설명

첫 번째 줄에 자연수 N(1 ≤ N ≤ 1,000), M(1 ≤ M ≤ N)이 주어진다. 

다음 줄에는 가수 BBB가 라이브에서 부른 순서대로 곡의 길이가 분 단위로 주어진다. (단, 곡의 길이는 자연수이며 10,000분을 넘지 않음)

 

*출력 설명

첫 번째 줄부터 DVD의 최소 용량 크기를 출력하시오.

 

 

---

 

 

풀이(Python)

답안

import sys
sys.stdin = open('AA/input_24.txt', 'rt')

def Count(capacity):
    cnt = 1
    sum = 0
    for x in Music:
        if sum + x > capacity:
            cnt += 1
            sum = x
        else:
            sum += x
    return cnt

n, m =map(int, input().split())
Music = list(map(int, input().split()))
maxx = max(Music)

lt = 1
rt = sum(Music)
while lt <= rt:
    mid = (lt + rt) // 2
    if mid >= maxx and Count(mid) <= m:
        res = mid
        rt = mid - 1
    else:
        lt = mid + 1
print(res)

# 출력 : 17

 

input_24.txt(입력)

9 3
1 2 3 4 5 6 7 8 9

 

 

중요내용

1. 이분 검색의 원리를 활용한 코드이다.
2. 변수 cnt의 초기 설정값이 1인 것은 노래를 녹음하기 위해 최소 1개의 DVD가 필요하기 때문이다.
3. DVD의 최소 용량은 반드시 길이가 가장 긴 노래의 용량보다는 커야 한다.
4. 변수 maxx는 n개의 노래 중 길이가 가장 긴 노래의 용량 크기를 할당 받는다.