Cod sursa(job #2924725)

Utilizator AnSeDraAndrei Sebastian Dragulescu AnSeDra Data 9 octombrie 2022 15:49:03
Problema Transport Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.99 kb
#include <fstream>

using namespace std;

#define Nmax 16000
#define VolSalteleMax 16000

int v[Nmax];

bool sePoate(long long nr, int cntmax, int n)
{
    int i, cnt;
    long long s;

    s = 0;
    cnt = 0;
    for(i = 0; i < n && cnt < cntmax; i++)
    {
        if(s + v[i] > nr)
        {
            s = 0;
            cnt++;
        }
        s += v[i];
    }

    if(cnt >= cntmax)
    {
        return 0;
    }

    return 1;
}

int main()
{
    ifstream fin("transport.in");
    ofstream fout("transport.out");

    int n, k, i;
    long long l, r, mid, sol;

    fin >> n >> k;

    for(i = 0; i < n; i++)
    {
        fin >> v[i];
    }

    l = 1;
    r = Nmax * VolSalteleMax;

    while(l <= r)
    {
        mid = (l + r) / 2;

        if(sePoate(mid, k, n))
        {
            sol = mid;
            r = mid - 1;
        }
        else
        {
            l = mid + 1;
        }
    }

    fout << sol;

    return 0;
}