[코딩테스트] 뮤직비디오(결정알고리즘)

문제

지니레코드에서는 불세출의 가수 조영필의 라이브 동영상을 DVD로 만들어 판매하려 한다. DVD에는 총 N개의 곡이 들어가는데, DVD에 녹화할 때는 라이브에서의 순서가 그대로 유지되어야 한다. 순서가 바뀌는 것을 우리의 가수 조용필씨가 매우 싫어한다. 즉, 1번 노래와 5번 노래와 같은 DVD에 녹화하기 위해서는 1번과 5번 사이의 모든 노래도 같은 DVD에 녹화해야한다.

 

지니레코드 입장에서는 이 DVD가 팔릴 것인지 확신할 수 없기 때문에 이 사업에 낭비되는 DVD를 가급적 줄이려고 한다. 고민 끝에 지니레코드는 M개의 DVD에 모든 동영상을 녹화하기로 하였다. 이 때 DVD의 크기(녹화 가능한 길이)를 최소로 하려고 한다. 그리고 M개의 DVD는 모두 같은 크기여야 제조원가가 적게 들기 때문에 꼭 같은 크기로 해야 한다.

 

입력설명

첫째 줄에 자연수 N, M이 주어진다. 다음 줄에는 조영필이 라이브에서 부른 순서대로 부른 곡의 길이가 분 단위로(자연수) 주어진다. 부른 곡의 길이는 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에 모든 영상을 녹화할 수 없다.

 

처음 생각 했던 답안..

# DVD 용량이 최소화 되어야 함.

n, m = 9, 3
vedio = [1,2,3,4,5,6,7,8,9]

lt = 1
rt = sum(vedio) # 한 장만에 다 담는 다 쳐도 되기 때문이다.

base=(lt+rt)//2 # 반쪽으로 가르기

while lt<rt:
    SUM = 0
    cnt = 0
    for i in vedio: # base(한 dvd당 담을 수 있는 분) 개수세기
        SUM +=i
        if SUM>=base:
            cnt+=1
            SUM=0
    if m<=cnt:
        lt = base+1
    else:
        rt = base-1
        
    base=(lt+rt)//2 # 반쪽으로 가르기

    print(lt,rt,base)


print(base)

 

mid 설정도 굳이 처음에 안해도 되었었고, 그리고 개수를 세는 과정에서 sum = 0 으로 초기화해서는 안되었다.

sum = i 로 설정해야 다음 SUM을 구할 때도 제대로 나오는군..

그리고 res 변수를 따로 설정해서 마지막 mid 값을 들고와 야하는구나..~

 

강의 답안

# DVD 용량이 최소화 되어야 함.


def Count(capacity):# mid(한 dvd당 담을 수 있는 분) 개수세기
    cnt = 1
    sum = 0
    for x in vedio: #Music 리스트에 있는 노래들을 하나하나 접근할 것.
        if sum+x>capacity:
            cnt+=1
            sum=x
        else:
            sum+=x
    return cnt
        
n, m = 9, 3
vedio = [1,2,3,4,5,6,7,8,9]

lt = 1
rt = sum(vedio) # 한 장만에 다 담는 다 쳐도 되기 때문이다.

res = 0

while lt<rt:
    mid=(lt+rt)//2 # 반쪽으로 가르기

    if Count(mid)<=m:
        res = mid
        rt = mid-1
    else:
        lt = mid+1
    print(lt,rt,mid)


print(res)

 

반례 수정

9 9 

1 2 3 4 5 6 7 8 9

DVD 하나의 용량은 최소한 9보다는 크고 같아야 한다는 것.

노래중에 가장 긴 곡을 담을 수 있는 opacity가 존재해야함.

어떤 노래보다도 opacity가 크거나 같아야 된다는 논리가 있다.

 

# DVD 용량이 최소화 되어야 함.


def Count(capacity):# mid(한 dvd당 담을 수 있는 분) 개수세기
    cnt = 1
    sum = 0
    for x in vedio: #Music 리스트에 있는 노래들을 하나하나 접근할 것.
        if sum+x>capacity:
            cnt+=1
            sum=x
        else:
            sum+=x
    return cnt
        
n, m = 9, 3
vedio = [1,2,3,4,5,6,7,8,9]
maxx = max(vedio)

lt = 1
rt = sum(vedio) # 한 장만에 다 담는 다 쳐도 되기 때문이다.

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(lt,rt,mid)


print(res)