[백준] 2343번 기타 레슨 - Python / 알고리즘 중급 1/3 - 이분 탐색 (연습)

ByungJik_Oh·2025년 7월 14일

[Baekjoon Online Judge]

목록 보기

195/244

💡 문제

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

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

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

입력

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

출력

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

💭 접근

이 문제는 이분 탐색을 활용한 매개 변수 탐색 문제로, 블루레이 하나의 길이를 임의로 설정한 뒤, 입력으로 주어진 m개의 블루레이를 만들 수 있는지 판단하는 문제이다.

따라서 변수는 다음과 같이 설정했다.

start : 강의 중 가장 긴 강의의 길이 → max(lectures)
end : 모든 강의를 더한 길이 → sum(lectures)
mid : 실제 블루레이 하나의 길이 → (start + end) // 2

변수를 위처럼 설정하고, mid값을 조절하며 블루레이 m개를 만들기 위한 최소한의 mid 값을 찾으면 된다.

따라서 mid 길이만큼 블루레이를 만들었을 때 블루레이의 개수가 m보다 적거나 같다면 블루레이 길이를 더 줄일 수 있는 여지가 있는 것이므로 mid를 줄인다.

반대로, mid 길이만큼 블루레이를 만들었을 때 블루레이의 개수가 m보다 많다면 블루레이의 길이를 늘려 개수를 줄여야 하므로 mid를 늘린다.

📒 코드

def binary_search():
    # 블루레이 하나의 길이는 최소한 영상의 길이의 최대보다 같거나 커야한다.
    start = max(lectures)
    # 블루레이 하나의 길이는 모든 영상의 길이의 합보다 클 순 없다.
    end = sum(lectures)

    ans = 0
    while start <= end:
        mid = (start + end) // 2

        cnt = 0
        tmp = 0
        for lecture in lectures:
            if tmp + lecture > mid:
                cnt += 1
                tmp = lecture
            elif tmp + lecture == mid:
                cnt += 1
                tmp = 0
            else:
                tmp += lecture

        if tmp != 0:
            cnt += 1
        
        if cnt <= m:
            ans = mid
            end = mid - 1
        else:
            start = mid + 1
    return ans

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

print(binary_search())