문제설명
지니레코드에서는 불세출의 가수 조영필의 라이브 동영상을 DVD로 만들어 판매하려 한다. DVD에는 총 N개의 곡이 들어가는데, DVD에 녹화할 때에는 라이브에서의 순서가 그대로 유지 되어야 한다. 순서가 바뀌는 것을 우리의 가수 조영필씨가 매우 싫어한다. 즉, 1번 노래와 5번 노래를 같은 DVD에 녹화하기 위해서는 1번과 5번 사이의 모든 노래도 같은 DVD에 녹화해야 한다. 또한 한 노래를 쪼개서 두 개의 DVD에 녹화하면 안된다.
지니레코드 입장에서는 이 DVD가 팔릴 것인지 확신할 수 없기 때문에 이 사업에 낭비되는 DVD를 가급적 줄이려고 한다. 고민 끝에 지니레코드는 M개의 DVD에 모든 동영상을 녹화하기 로 하였다. 이 때 DVD의 크기(녹화 가능한 길이)를 최소로 하려고 한다. 그리고 M개의 DVD는 모두 같은 크기여야 제조원가가 적게 들기 때문에 꼭 같은 크기로 해야 한다.
문제해결 아이디어
- 최종적으로 구해야하는 값은 dvd의 최소용량크기
- n개의 곡을 m개의 dvd에 순서대로 넣어야함(이분검색 활용)
- 반례조건 추가:가장 용량이 큰 노래가 하나의 DVD용량보다는 크거나 같아야함
- 해당 용량의 DVD선택시 전체 노래를 몇개의 DVD에 담을 수 있는지 Count()함수를 별도로 만들어서 확인
코드
import sys
base='/Users/leesh970930/Desktop/pythonalgorithm_formac/섹션 4/'
sys.stdin=open(base+'3. 뮤직비디오/in2.txt', "r")
n,m=map(int,input().split())
arr=list(map(int,input().split()))
res=0
lt=1
rt=sum(arr)
maxx=max(arr)
#해당 용량으로 노래를 몇개의 CD에 나누어 담을 수 있는지 확인
def Count(capacity):
cnt=1
sum=0
for x in arr:
if sum+x>=capacity: #sum(누적용량)+현재노래(x)를 더한 값이 capacity를 초과하는지 확인.
#초과하면 다음 CD에 저장해야함
cnt+=1
sum=x #CD의 갯수를 늘려준 뒤, 누적용량을 현재 노래(x)로 다시 초기화
else:
sum+=x
return cnt
while lt<=rt:
mid=(lt+rt)//2
if Count(mid)<=m and mid>=maxx:#반례조건 추가:가장 용량이 큰 노래가 하나의 DVD용량보다는 크거나 같아야함
res=mid
#조건에 부합하는 더 작은 mid값이 존재하는지 확인해야함 ->작은 값들 중 확인
rt=mid-1
else:#용량이 작으므로 더 큰 쪽에서 확인해아함
lt=mid+1
print(res)
