Cod sursa(job #2331118)

Utilizator 53215343214Nejneriu Mihai 53215343214 Data 29 ianuarie 2019 10:49:19
Problema Transport Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.71 kb
#include <iostream>
#include <fstream>
using namespace std;

ifstream f("transport.in");
ofstream g("transport.out");

int n, k, a[16004], sum, maxi;
int main()
{
    f>>n>>k;
    for(int i=1; i<=n; i++)
        {f>>a[i];
        maxi=max(a[i],maxi);
         sum=sum+a[i];}
    int st,dr,mij,i,cnt,nr=0;
    st=maxi;
    dr=sum;
    while(st<=dr)
    {
        mij=(st+dr)/2;
        cnt=0;
        sum=0;
        for(i=1; i<=n; i++)
            {
                if(sum+a[i]<=mij&&a[i]<=mij) sum=sum+a[i];
                    else {cnt++; sum=a[i];}
            }
            if(sum<=mij) cnt++;
            if(cnt>k) st=mij+1;
        else if(cnt<=k) dr=mij-1, nr=mij;
    }
    g<<nr;
}