BOJ 2343 기타 레슨

문제

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

주어진 수들을 m개로 쪼갤 때 각 CD의 수들의 합의 최댓값을 최소화 시키는 문제이다.

 

풀이

문제는 매우 간단한 파라마트릭 서치 문제이다.

결정함수에서는 각 CD의 용량이 time 일때 m개 이하로 배정할수 있는지 여부를 반환하면 된다.

당연히 결정함수는 넣을 수 있는것은 최대한 쑤셔 넣는 그리디이다.

 

이 문제가 나한테 어려웠던 이유는 바로!!!!!! 구현이다.

그리디 부분에서 몇개 값을 건너뛴다거나, 무한루프를 돈다거나 하는 문제가 발생해 한동안 애먹었었는데, 

바로 time<arr[i] 인 특수한 경우이다.

일단 time<arr[i] 인경우가 나오면 바로 return false; 를 해주고, 나머지 경우에는 포문을 한번씩 돌면서 계산하다.

여기서도 함정이 있는데, 마지막에 cnt값이 올라가지 못하므로, 처음부터 초기값을 1로 설정해 주는 센스가 필요하다.

 

시간 복잡도 : $O(Nlog10^9)$

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

typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;

const int MAXN = 1e5;

int n, m, arr[MAXN+10];

bool decide(int time)
{
    int i;
    int sum=0, cnt=1;
    for(i=0; i<n; i++)
    {
        if(arr[i]>time) return false;
        if(sum+arr[i]>time)
        {
            sum=0;
            cnt++;
        }
        sum+=arr[i];
    }

    return cnt<=m;
}

int main()
{
    cin.tie(0); cout.tie(0);
    ios_base::sync_with_stdio(false);
    int i, j;

    scanf("%d%d", &n, &m);
    for(i=0; i<n; i++) scanf("%d", &arr[i]);

    int lo=0, hi=1e9+10;
    while(lo+1<hi)
    {
        int mid=lo+hi>>1;
        if(decide(mid)) hi=mid;
        else lo=mid;
    }

    printf("%d", hi);
    return 0;
}