결정알고리즘 2

 

지니레코드에서는 불세출의 가수 조영필의 라이브 동영상을 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의 최소 용량 크기를 출력하세요.

▣ 입력예제

9 3 1 2 3 4 5 6 7 8 9

▣ 출력예제
17

import sys
sys.stdin=open("input.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)
res=0
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) 

n개의 곡을 모두 같은 m개의 크기 만들어 최소 용량 크기(res)를 출력하면 되는 문제입니다.

DVD한장의 최소 용량이 1이기 때문에 lt=1, rt는 n(9)개의 곡을 모두 합한 sum으로 선언합니다.

그리고 mid의 값은 저장하고자 하는 dvd의 용량 중 가장 큰 것의 용량보다 크다는 조건이 있어야 9도 포함할 수 있기 때문에 maxx라는 조건을 추가합니다.

최소 용량을 저장할 res를 0으로 초가화 한 뒤, while문을 돌립니다.

중간 값을 나눠서 탐색을 해야하기 때문에, lt와 rt를 더한 뒤, 2로 나눠준다면 총 용량의 중간값을 도출합니다.

if문을 실행하기 전 count라는 정의함수를 사용하기 위해 def로 정의해주겠습니다.

count는 n개의 곡을 모두 저장하면 dvd 몇장이 필요한지 return해주는 함수입니다.

cnt=1은 dvd 한 장이 필요, sum=0은 저장되는 노래를 나타냅니다.

Music이라는 list에 있는 노래들에 접근하고, 노래의 시간(x)만큼 반복합니다.

sum+x>capacity가 된다면, 용량을 초과했다는 의미입니다. 그렇게 되면 새로운 dvd가 필요한 것이기 때문에 cnt+=1이 되고 sum에 노래의 시간을 누적합니다. 초과되지 않다면(else) sum에 계속 누적하면 됩니다. x만큼 반복한 뒤, cnt를 아래의 count 함수에 return하게됩니다.

return 받은 count함수가 m보다 작게되면, 최소 용량(res)에 mid값을 누적합니다. 그리고 작은 범위로 좁혀나가야 되기 때문에, rt를 mid-1로 줄이고 다시 반복문이 돌아가게 됩니다.

답이 아니라면, 용량이 너무 작아서 더 큰 용량이 필요한 것입니다.

즉 lt=mid+1이 되어 더 큰 범위에서 search하게 됩니다.

출력하게 되면 답이 나옵니다.

17