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

문제:

뮤직비디오(결정알고리즘)
지니레코드에서는 불세출의 가수 조영필의 라이브 동영상을 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


설명 : 3개의 DVD용량이 17분짜리이면 (1, 2, 3, 4, 5) (6, 7), (8, 9) 이렇게 3개의 DVD로 녹음을 할
수 있다. 17분 용량보다 작은 용량으로는 3개의 DVD에 모든 영상을 녹화할 수 없다.

내 풀이(틀린 풀이)

n,m=map(int, input().split())
length_list=list(map(int, input().split()))
length_list.sort()

min=sum(length_list)//m
max=sum(length_list)


while min<=max:
    count=0
    mid=(min+max)//2
    for i in range(n):
        if mid<0:
            count+=1
            mid=(min+max)//2
        mid-=length_list[i]
    
    if count>=3:
        min=mid+1
    
    elif count<3:
        max=mid-1

print(mid)

-수록된 곡을 묶음으로 분류하는 함수를 작성하는 것을 실패함

 

 

강의 풀이.

def count(capacity):
    cnt=1
    #cnt를 1로 두느냐 0으로 두느냐-->시작 포인트부터 세느냐 아니면 끝나는 포인트마다 세느냐의 차이
    #이를 루프가 끝날 때 개수가 반영되는지를 따져야 됨--만약 안되면 시작 포인트부터 세야됨
    sum=0
    for i in Music:
        if sum+i>capacity:
            sum=i
            cnt+=1
        else:
            sum+=i
    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:
    #mid>=maxx-->리스트에서 가장 길이가 긴 음악의 길이보다는 커야됨!(곡이 아예 안들어가지면 안됨)
        res=mid
        rt=mid-1
    else:
        lt=mid+1
print(res)

-def count(capacity)에서

cnt=1로 둘지 cnt=0으로 두어야 하는지를 알지 못했음

--> 루프가 끝날 때 횟수가 반영되는 지를 체크해야 됨

만약 안되면 시작 포인트부터 수를 세야함 cnt=1로 두어야 함

끝날 때 반영된다면 cnt=0

 

-주목해야 될 코드

def count(capacity):
    cnt=1
    #cnt를 1로 두느냐 0으로 두느냐-->시작 포인트부터 세느냐 아니면 끝나는 포인트마다 세느냐의 차이
    #이를 루프가 끝날 때 개수가 반영되는지를 따져야 됨--만약 안되면 시작 포인트부터 세야됨
    sum=0
    for i in Music:
        if sum+i>capacity:
            sum=i
            cnt+=1
        else:
            sum+=i
    return cnt

-sum+i>capacity 일 때 sum=i로 넘김

계속 for문이 돌아야 되는데, sum+i가 초과하는 것을 알았을 때의 i를 다시 합계로 반영해야 된다

그래서 sum=i로 넘겼음-->이 부분을 생각해내지 못함

 

-sum+i가 초과하지 않았을 경우 계속 합계에 반영하였음

 

나머지 이분 검색(결정)으로 해결