백준

[2343] 기타 레슨 (JAVA)

댈팽이 2024. 3. 30. 17:39
728x90

# 문제 설명

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

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

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

입력

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

출력

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

 

# 정답 코드

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());

        int N = Integer.parseInt(st.nextToken()); //강의의 수
        int M = Integer.parseInt(st.nextToken()); //블루레이의 수

        int[] arr = new int[N];
        int start = 0; //강의 시간 최댓값
        int end = 0; //모든 강의 시간 합

        st = new StringTokenizer(br.readLine());
        for (int n = 0; n < N; n++) {
            arr[n] = Integer.parseInt(st.nextToken());
            start = Math.max(start, arr[n]);
            end += arr[n];
        }

        while (start <= end) {
            int middle = (start + end) / 2;
            int cnt = 0; //필요한 블루레이 개수
            int sum = 0; //강의 시간 합

            for (int i = 0; i < arr.length; i++) {
                if (sum + arr[i] > middle) {
                    cnt++;
                    sum = 0;
                }

                sum += arr[i];
            }

            //sum이 0이 아니면 블루레이가 하나 더 필요함
            if (sum != 0) {
                cnt++;
            }

            if (cnt > M) {
                start = middle+1;
            }
            else {
                end = middle-1;
            }
        }

        //가능한 블루레이 크기 중 최솟값을 출력
        System.out.print(start);
    }
}

 

가능한 블루레이 크기 중 최솟값을 찾기 위해 이진 탐색을 사용하였다.

블루레이 크기의 범위는 가장 긴 강의 시간 ~ 모든 강의 시간의 합이다.

따라서 start는 가장 긴 강의 시간, end는 모든 강의 시간의 합으로 설정한다.

 

중간값 middle을 구해 가능한 블루레이 크기인지 검사한다.

사용한 블루레이 개수 cnt와 강의 시간 합 sum을 사용한다.

 

강의 시간 배열 arr을 순회하면서 sum에 현재 강의 시간을 더한다.

만약 sum에 현재 강의 시간을 더한 값이 middle보다 크면, cnt를 1 증가하고 sum을 0으로 초기화한다.

배열 순회가 끝나고 sum이 0이 아니면 블루레이가 하나 더 필요하므로 cnt를 1 증가한다.

 

구한 cnt가 블루레이 개수 M보다 크면, middle을 늘려야하므로 start를 middle+1로 설정한다.

cnt가 M보다 작거나 같으면, middle을 줄여야하므로 end를 middle-1로 설정한다.

 

이진 탐색이 끝나면 가능한 블루레이 크기 중 최솟값인 start를 출력한다.

728x90