[python] 백준 - 2343. 기타 레슨

🤔문제 해결

S1 | 이분 탐색

이분 탐색 문제는  무엇을 탐색  할 것인지가 가장 중요합니다. 

이 문제에서는  블루레이의 최소 크기  를 찾아야 합니다.

 

처음에 각각의 블루레이에 레슨들을 순서대로 담아야 합니다.

블루레이의 크기가 11라고 가정해 보겠습니다. 그렇게 되면 레슨들의 합이 11을 넘어서는 안됩니다.
그럼 아래와 같은 결과를 얻을 수 있습니다.

총 5개의 블루레이에 담아야 합니다. 테스트 케이스에서는 3개에 담으라고 했으니 답이 될 수 없습니다.

 

그렇다면 블루레이 크기를 늘려서 한 블루레이에 좀 더 많이 담아야 겠다는 생각을 할 수 있습니다.

15라고 가정해 보겠습니다.

총 4개의 블루레이에 담았습니다. 하지만 여전히 블루레이 개수가 많습니다. 블루레이의 크기를 더 늘려야 합니다.

 

만약 30으로 크기를 늘리면 어떻게 될까요?

블루레이가 2개 밖에 나오지 않아 이것도 답이 될 수 없습니다.

 

이렇게 블루레이의 개수가 많으면 블루레이 크기를 늘리고, 블루레이 개수가 적으면 블루레이 크기를 줄여가며 이분탐색 을 하면 됩니다.

 

그렇다면 여기서 블루레이의 크기를 정하는게 중요합니다. 블루레이를 하나에 담는 크기는 레슨들의 합입니다.

여기서 블루레이가 45(레슨들의 합)이상이면 블루레이 하나에 모든 레슨을 담을 수 있습니다.

또, 블루레이가 최소 9(레슨들 중 제일 큰 값)는 되어야 블루레이에 모든 레슨을 담을 수 있습니다.

이렇게 왼쪽값은 9, 오른쪽 값은 45로 두고, 두 값의 중간값((9+45)//2)을 블루레이의 크기로 시작합니다.

---

• 먼저 왼쪽, 오른쪽, 중간값을 초기화 합니다. (위에서 설명)
• 블루레이 크기에 따라서 레슨들의 값을 차례차례 더해서 넣어보고 그 크기를 넘어갈 때마다 블루레이 개수를 +1씩 해줍니다.
• 이제 블루레이의 개수에 따라 크기를 늘릴지 줄일지 정합니다.
  • 블루레이 개수가 부족하다면 right = mid - 1 을 해줘서 블루레이 크기를 줄입니다.
  • 블루레이 개수가 넘어간다면 left = mid + 1 을 해줘서 블루레이 크기를 늘립니다.
• 마지막으로 left 를 출력하면 답이 됩니다.

 

💻소스 코드

def add_lesson():
    cnt = 0  # 레코드 갯수 세기
    sum_lesson = 0  # 한 레코드에 들어갈 레슨들의 합
    for i in range(N):
        if sum_lesson + lesson_list[i] > mid:
            cnt += 1
            sum_lesson = 0

        sum_lesson += lesson_list[i]
    else:
        if sum_lesson:
            cnt += 1
    return cnt


if __name__ == "__main__":
    N, M = map(int, input().split())  # N: 레슨 수, M: 블루레이 수
    lesson_list = list(map(int, input().split()))  # 레슨들

    right = sum(lesson_list)   # 레슨을 하나의 레코드에 다 담을 수 있을 때 레코드의 크기는 레슨의 합이다
    left = max(lesson_list)  # 레코드가 가질 수 있는 가장 작은 크기
    while left <= right:
        # 레코드 크기 중간값 구하기
        mid = (left + right) // 2
        cnt = add_lesson()
        if cnt <= M:  # 레코드 숫자가 모자라면 레코드 크기(mid)를 줄인다.
            right = mid - 1
        elif cnt > M:  # 레코드 숫자가 더 많아지면 레코드 크기(mid)를 늘린다
            left = mid + 1

    # 답은 left 에 있다. (최소 크기)
    print(left)

 

 

📕문제 확인

출처: BACKJOON ONLINE JUDGE

링크: https://www.acmicpc.net/problem/2343

2343번: 기타 레슨

---

---

---

문제

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

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

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

입력

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

출력

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

힌트

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

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