[BOJ / Node.js] 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분을 넘지 않는다.

출력

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

예제 입력 1

9 3
1 2 3 4 5 6 7 8 9

예제 출력 1

17

 

Solve

• 주어진 범위 내에서 최소값을 구하는 최적화 문제로 파라메트릭 서치를 사용하였습니다.
• 조건에 따르면 최소 한 개의 동영상을 블루레이에 담을 수 있어야 하기 때문에, 가장 작은 범위값인 left를 주어진 배열 n의 Max값으로 설정하였습니다. ex) 예제 입력으로 예시를 들면 최소 한개의 동영상을 넣을 수 있는 블루레이의 크기는 9입니다. 그보다 작다면, 9는 조건에 부합하지 않아 블루레이에 담을 수 조차 없습니다.
• 가장 큰 범위 값 right는 모든 동영상들을 하나의 블루레이에 담을 수 있는 배열의 합으로 설정하였습니다.
• 현재 탐색중인 크기(mid)를 기준으로, 동영상들의 길이를 구해서 blueray의 갯수를 구하고, blueray의 개수보다 크다면 범위값을 키우고, 작거나 같다면 범위값을 줄입니다.

Problem

• 범위값을 조정하는 과정에서 blueray의 갯수가 m과 일치하다면 올바른 탐색범위이기 때문에 right의 범위에 mid도 포함되어야 했지만 이분탐색과 과정이 헷갈려 right = mid - 1로 설정하는 바람에 오답이 나왔었습니다..

Code

const input = require("fs").readFileSync("/dev/stdin").toString().trim().split("\n");

const [n, m] = input[0].split(' ').map(el=>+el);
const arr = input[1].split(' ').map(el=>+el);

let left = Math.max(...arr);
let right = arr.reduce((prev,cur)=>prev+cur);
let mid = Math.floor((left + right) / 2)
while(left < right) {

    let blueray = 1; let sum = 0;
    for(let i = 0 ; i < n; i++) {
        sum += arr[i];
        if(sum > mid) {
            blueray += 1;
            sum = arr[i];
        }
    }

    if(blueray <= m) {
        right = mid;
    } else {
        left = mid + 1;
    }
    mid = Math.floor((left + right) / 2)
}

console.log(mid);