백준 2343번 C++ 풀이

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

2343번: 기타 레슨

시간 제한	메모리 제한	solved.ac 티어
2초	128MB	실버 1

문제

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

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

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

입력

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

출력

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

---

 이분탐색으로 풀이해야 하는 문제이다. (최대 강의의 수) * (최대 강의의 길이) = 10^9이기 때문에 단순한 반복문으로는 시간초과가 날 것이다.

 

이 문제를 이분탐색으로 풀 때, 시작값과 끝값을 어떻게 잡는지가 조금 애매했는데, 끝값의 경우는 모든 강의의 시간을 합한 값으로 초기화가 되어야 하고, 시작값은 모든 강의의 시간중에서 제일 큰 값으로 초기화가 되어야 한다. 왜냐하면 적어도 모든 강의가 레코드 하나에는 들어가야 하기 때문이다.

 

#include <bits/stdc++.h>
using namespace std;

vector<int> v;

int blue_ray_count(int amount){
    int sum = 0;
    int cnt = 1; // 기본적으로 하나에는 담기기 때문에 cnt를 1로 초기화
    for(int i = 0; i < v.size() - 1; i++){
        sum += v[i];
        if(sum + v[i + 1] > amount){ // 꽉 채워졌다면
            sum = 0; // 초기화
            ++cnt; // 개수 증가
        }
    }
    return cnt;
}

int main () {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    int n, m;
    int start = -1;
    int end = 0;
    int mid, result, temp;
    
    cin >> n >> m;

    for(int i = 0; i < n; i++) {
        cin >> temp;
        v.push_back(temp);
        end += temp; // end의 초기값은 모든 값의 합
        start = max(temp, start); start의 초기값은 모든 값중에서 제일 큰 값.
    }

    while(start <= end){ // 이분탐색
       mid = (start + end) >> 1;
       
       if(blue_ray_count(mid) <= m) { // 개수가 적다면 
           end = mid - 1;
           result = mid;
       }
       else {
           start = mid + 1;
       }
    }
    cout << result;
}