[백준] # 2343. 기타 레슨 (파이썬)

문제 링크

 

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분을 넘지 않는다.

출력

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

 

IDEA

 

처음엔  N개를 M개로 어떻게 나누지?를 고민하다가 M-1개의 구분 칸을 조합을 사용해서 만들고 최소 크기를 구하자!로 구현했지만 메모리 초과가 떴다. 힌트를 보니 이분탐색이었는데 보고 한참을 생각했다. M개로 나눠야 하는데 어떻게 이분탐색을 적용하지? 생각하다가 생각을 전환해서 가능한 블루레이 크기는 sum(N)//M보다 크고 그 수부터 N까지 이분탐색을 실시해서 가능한 최소 블루레이 크기를 구하면 될 것이라 생각해서 풀었더니 풀렸다.

 

CODE

파이썬 조합 라이브러리를 활용해서 풀려다가 메모리 초과로 실패했다.

# 2343. 기타 레슨 - 실버1
# 메모리 초과!!
# n-1개중 m-1개를 뽑아서 구분 칸을 만든다.
from itertools import combinations
n, m = map(int,input().split())
# item에 1~n-1까지 저장 후 조합을 이용해 m-1개 만큼 뽑아 새로운 리스트 생성
items = [i for i in range(1,n)]
division = list(combinations(items,m-1))
lesson_list = list(map(int,input().split()))
min_size = sum(lesson_list)
for div_list in division:
    for i,val in enumerate(div_list):
        if i == 0:
            size = sum(lesson_list[0:val])
        else:
            size = max(size, sum(lesson_list[div_list[i-1]:val]))
    size = max(size, sum(lesson_list[val::]))
    min_size = min(min_size,size)
print(min_size)

 

 

힌트를 본 후 이분탐색으로 구현한 알고리즘이다.

# 2343. 기타 레슨 - 실버1 / 2번째 시도
# 이분탐색
def binarySearch(left,right):
    result = size
    while left <= right:
        mid = (left + right) // 2
        if mid < max(lesson_list):
            left = mid + 1
            continue
        count = 0
        temp = 0
        for i in range(n):
            temp += lesson_list[i]
            if temp > mid:
                temp = lesson_list[i]
                count += 1
        if count < m:   # 가능한 경우
            right = mid - 1
            result = min(mid,result)
        else:
            left = mid + 1
        #print(count, left, right,mid)
    return result
n, m = map(int,input().split())
lesson_list = list(map(int,input().split()))
size = sum(lesson_list)
#print(size)
print(binarySearch(size//m,size))

 

사용 알고리즘

 

이분 탐색

 

풀이 평

 

아마 힌트가 없었다면 풀기 힘들었을 것 같다. 아직 문제를 보고 무슨 알고리즘을 써야하는지 정확히 판단할 수 있는 경지에는 못 오른 것 같다. 더 열심히 해야지.