Cod sursa(job #2528702)

Utilizator robertanechita1Roberta Nechita robertanechita1 Data 22 ianuarie 2020 13:38:46
Problema Transport Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.88 kb
#include <bits/stdc++.h>

using namespace std;
int a[16003], n, k;
ifstream fin("transport.in");
ofstream fout("transport.out");
int Test(int C)
{
    ///ret 1 daca pot imparti vectorul in cel mult k secvente
    ///fiecare secventa avand suma <= C
    int s, cnt, i;
    s=0;
    cnt=1;
    for(i=1; i<=n; i++)
    {
        if(a[i]>C) return 0;
        if((s+a[i])<=C) s+=a[i];
        else
        {
            s=a[i];
            cnt++;
        }

    }
    return (cnt<=k);
}
int CB()
{
    int st, dr, mij, p;
    st=1;
    dr=256000000;
    p=1;
    while(st<=dr)
    {
        mij=(st+dr)/2;
        if(Test(mij)==1)
        {
            p=mij;
            dr=mij-1;
        }
        else st=mij+1;
    }
    return p;
}
int main()
{
    fin >> n >> k;
    for(int i=1; i<=n; i++)
        fin >> a[i];
    fout << CB();

    return 0;
}