인프런 커뮤니티 질문&답변

김남균님의 프로필 이미지
김남균

작성한 질문수

파이썬 알고리즘 문제풀이 입문(코딩테스트 대비)

뮤직비디오 코드 부분

작성

·

147

0

import sys
sys.stdin = open("input.txt", "r")

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()))
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)


답변 2

1

김태원님의 프로필 이미지
김태원
지식공유자

만약 남균님의 생각대로 만약 cnt=0으로 초기화하고 노래가 한곡만 있다고 생각해보세요.  노래 한곡을 저장해야 하니까 DVD 한장이 필요해 Count함수는 1을 리턴해야 하지만 0을 리턴합니다.

Count 함수가 필요한 DVD개수를 세는 방식은 다음과 같습니다.

1. 먼저 빈 DVD가 한장 있다.

2. 이 DVD에 노래를 하나씩 담아본다.

3. 노래를 담아보다가 더 이상 넣을 수 없으면 빈 DVD를 하나 준비한다. 이때 cnt가 1증가합니다.

이 과정을 반복하는 것입니다.

0

김남균님의 프로필 이미지
김남균
질문자

뮤직비디오 문제에서요 def Count() 함수에서 cnt를 1로 초기화하고 시작하는게 잘 이해가 가지 않습니다.

capacity를 넘을 때 1씩 증가시키는 것으로 되어 있는데, 그렇다고 한다면 처음에는 0으로 시작해야 하는게 아닌가 하는 생각이 자꾸 듭니다... 제가 뭔가 놓친거 같은데 생각해볼 수록 헷갈려서요. 

김남균님의 프로필 이미지
김남균

작성한 질문수

질문하기