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

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

내 답은 틀렸다. 이분탐색을 통해 해설과 비슷한 방식으로 코드를 짰지만 Count 함수에서 잘못 구현한 부분이 있었다.

<내 답안>

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

def Count(capa):
    s = 0
    cnt = 0
    for i in range(n):
        s += li[i]
        if s > capa:
            cnt += 1
            s = li[i]

    print(cnt)
    return cnt

res = 0  # DVD 용량 크기 범위 : 1~sum(li) = 1~45
lt = 1
rt = sum(li)

while lt <= rt:
    mid = (lt+rt)//2
    if Count(mid) <= m:
        rt = mid - 1
        res = mid
    elif Count(mid) < m:
        rt = mid - 1
        res = mid
    else:
        lt = mid + 1

print(res)

res = mid 용량을 가지고 m개의 DVD에 담아낼 수 있는 방법을 구하는 부분부터 막혔다.
해설에 따르면 이것은 첫 곡부터 res를 넘지 않을 때까지 한 장, 그 다음 곡부터 더해가면서 총 m개의 DVD가 나오면 res 용량으로 m개의 DVD에 담아낼 수 있다는 것이다. 따라서 이때의 res는 답이 될 수 있다.
여기서 주의할 점은 만약 m개가 아닌 m개보다 작은 갯수에 담아낼 수 있다면 이때의 res도 당연히 답이 된다. m개보다 큰 갯수에 담아야 한다면 최소 m개가 필요하다는 것이니 이때의 res는 답이 될 수 없다.
즉, 답이 되는 res보다 큰 res(더 큰 용량)는 당연히 답이 된다는 점을 생각해야 한다. 따라서 이분탐색 범위에서 rt를 줄여나가야 한다.

<모범답안>

def Count(capacity):
    sum = 0
    cnt = 1
    for x in Music:
        if sum+x > capacity:  # x도 저장할 수 있는지 확인
            cnt += 1
            sum = x
        else:
            sum += x
    print(cnt)
    
    return cnt

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

lt = 1
rt = sum(Music)
res = 0

while lt <= rt:
    mid = (lt+rt)//2
    if Count(mid) <= m:
        res = mid
        rt = mid - 1
    else:
        lt = mid + 1

print(res)