2343번: 기타 레슨

기타 레슨

 

시간 제한           메모리 제한          제출          정답          맞힌 사람          정답 비율

2 초

128 MB

21703

7090

4988

30.978%

문제

강토는 자신의 기타 강의 동영상을 블루레이로 만들어 판매하려고 한다. 블루레이에는 총 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

힌트

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

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

 

 

유의할 점

• 블루레이는 최대 공간이다. => mid를 블루레이 크기라고 할때 값을 넘으면 새로운 블루레이에 채워넣어야한다.  

 

코드

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

int n,m, a[100004], s,ret=987654321;

bool check(int mid){
    for(int i=0;i<n;i++){ //블루레이 크기가 원소 크기보다 작으면 안된다. 크기를 빼면 계속 음수값 나옴
        if(a[i]>mid) return false;
    }
    int val=mid; int cnt=0;
    for(int i=0;i<n;i++){
        if(val-a[i]<0){ //새로운 블루레이 필요할때
            cnt++; //블루레이 개수 추가
            val=mid; //크기 채워넣기
        }
        val -= a[i];
    }
    if(val!=mid) cnt++; //마지막 블루레이
    return cnt<=m; //true든 false든 리턴
}
    
int main(){
    ios_base::sync_with_stdio(0); cin.tie(NULL); cout.tie(NULL);
    ///1. 입력받기
    cin >> n >> m; //n개의 블루레이, m개의 묶음
    for(int i=0;i<n;i++){
        cin >> a[i];
        s += a[i];
    }
    //2. 이분탐색
    int l=0; int r=s;
    int mid=0;
    while(l<=r){
        mid=(l+r)/2; //블루레이 크기
        if(check(mid)){ //묶음이 적으면 mid를 줄이기
            r=mid-1;
            ret = mid;
        } else l=mid+1;
    }
    //3. 정답 출력 
    cout << ret;
}

 

유의할 점

• 크기가 부족할때는 크기를 채우고 빼준다.

 int val=mid; int cnt=0;
    for(int i=0;i<n;i++){
        if(val-a[i]<0){ //새로운 블루레이 필요할때
            cnt++; //블루레이 개수 추가
            val=mid; //크기 채워넣기
        }
        val -= a[i];
    }

먼저 크기를 빼준 후 음수일때를 체크하면 음수일때의 원소가 블루레이에 들어가지 않는다.

 

• bool함수는 true이든 false이든 반환해주어야한다.
• 블루레이 크기는 블루레이 원소보다 커야한다(그렇지않으면 val-a[i]가 계속 음수가 나온다.

 for(int i=0;i<n;i++){ //블루레이 크기가 원소 크기보다 작으면 안된다. 크기를 빼면 계속 음수값 나옴
        if(a[i]>mid) return false;
    }
    int val=mid; int cnt=0;
    for(int i=0;i<n;i++){
        if(val-a[i]<0){ //새로운 블루레이 필요할때
            cnt++; //블루레이 개수 추가
            val=mid; //크기 채워넣기
        }
        val -= a[i];
    }