[백준] #2343 기타 레슨 - 파이썬

문제

강토는 자신의 기타 강의 동영상을 블루레이로 만들어 판매하려고 한다. 블루레이에는 총 N개의 강의가 들어가는데, 블루레이를 녹화할 때, 강의의 순서가 바뀌면 안 된다. 순서가 뒤바뀌는 경우에는 강의의 흐름이 끊겨, 학생들이 대혼란에 빠질 수 있기 때문이다. 즉, i번 강의와 j번 강의를 같은 블루레이에 녹화하려면 i와 j 사이의 모든 강의도 같은 블루레이에 녹화해야 한다.

강토는 이 블루레이가 얼마나 팔릴지 아직 알 수 없기 때문에, 블루레이의 개수를 가급적 줄이려고 한다. 오랜 고민 끝에 강토는 M개의 블루레이에 모든 기타 강의 동영상을 녹화하기로 했다. 이때, 블루레이의 크기(녹화 가능한 길이)를 최소로 하려고 한다. 단, M개의 블루레이는 모두 같은 크기이어야 한다.

강토의 각 강의의 길이가 분 단위(자연수)로 주어진다. 이때, 가능한 블루레이의 크기 중 최소를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 강의의 수 N (1 ≤ N ≤ 100,000)과 M (1 ≤ M ≤ N)이 주어진다. 다음 줄에는 강토의 기타 강의의 길이가 강의 순서대로 분 단위로(자연수)로 주어진다. 각 강의의 길이는 10,000분을 넘지 않는다.

출력

ㅅㅅ첫째 줄에 가능한 블루레이 크기중 최소를 출력한다.

예제 입력 1 

9 3
1 2 3 4 5 6 7 8 9

예제 출력 1 

17

힌트

강의는 총 9개이고, 블루레이는 총 3개 가지고 있다.

1번 블루레이에 1, 2, 3, 4, 5, 2번 블루레이에 6, 7, 3번 블루레이에 8, 9 를 넣으면 각 블루레이의 크기는 15, 13, 17이 된다. 블루레이의 크기는 모두 같아야 하기 때문에, 블루레이의 크기는 17이 된다. 17보다 더 작은 크기를 가지는 블루레이를 만들 수 없다.

---

복기

1. 이진탐색인 것은 파악했지만, start=0, end=max(ls)로 하면 안되었다.. 문제를 이해하고 설계하는 능력이 아직 부족하네.

리스트를 덩어리로 보고 부분 리스트의 합을 봐야한다. 

 

Step 1. start = 9, end = 45, mid = 27

이렇게 나눠주면 블루레이의 개수가 3보다 작으므로 mid를 더 감소시키고, cnt += 1를 해준다.

 

Step 2. start = 9, end = 26, mid = 17

블루레이 크기가 17보다 크면 안되므로 위의 그림과 같이 나눌 수 있다. 

블루레이의 개수도 3개이므로 정답이 될 수 있다. 

따라서, 정답 변수에 일단 저장해놓고 최솟값을 더 탐색하기 위해 end를 mid - 1로 설정해준다. 

 

Step 3. start = 9, end = 16, mid = 12 

이번엔 블루레이의 개수가 5개이므로 너무 많다. 

따라서 start를 mid + 1로 설정해서 덜 나눠지도록 해준다. 

 

Step 4. start = 13, end = 16, mid = 14 

step 3와 동일하므로 start를 mid + 1로 설정해준다. 

 

Step 5. start = 15, end = 16, mid = 15

여전히 블루레이의 개수가 3개보다 많으므로 start를 mid + 1로 설정해준다. 

 

Step 6. start = 16, end = 16, mid = 16 

step 5와 동일하므로 start를 mid + 1로 설정해준다. 

 

Step 7. start = 17, end = 16

start가 end보다 커졌으므로 종료한다. 

 

2. for loop에서 break가 없어도 되는 이유는, 

if total + ls[i] > mid:

에서 mid를 초과하면 cnt를 1 증가시키고 total 을 0으로 초기화하는데, 즉, 새로운 블루레이에 강의를 담는 것이다. 최종적으로 계산된 cnt 값으로 mid을 줄일지 늘릴지 결정하므로 break가 필요하지 않다. 

---

솔루션

N, M = map(int, input().split())
ls = list(map(int, input().split()))

start = max(ls)
end = sum(ls)

res = 0
while (start<=end):
    mid = (start+end)//2
    # 블루레이에 강의 넣기
    cnt, total = 1, 0
    for i in range(N):
        if total + ls[i] > mid:
            cnt += 1
            total = 0
        total += ls[i]
    if cnt <= M:
        res = mid
        end = mid-1
    else:
        start = mid+1

print(res)