Cod sursa(job #1078131)

Utilizator Aleks10FMI - Petrache Alex Aleks10 Data 12 ianuarie 2014 01:02:18
Problema Transport Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.56 kb
#include <iostream>
#include <fstream>

using namespace std;

int k,n,a[17000],minim=1<<19;
    int suma=0;
int cauta(int st, int dr){
    int mij=(st+dr)/2;
    int maxim=mij;
    int contor=1;
    for(int i=0;i<n;i++){
        //cout<<maxim<<'-'<<a[i]<<'\n';
        if(maxim-a[i]>=0)
            maxim=maxim-a[i];
        else if(maxim-a[i]<0){
            maxim=mij;
            contor++;
            maxim=maxim-a[i];
        }
    }
    //cout<<contor<<" "<<"("<<mij<<") ";
    if(contor<=k)
        minim=mij;
    if(contor<=k && st<mij){
        cauta(st,mij);
        //cout<<mij;
        /*if(contor<=minim)
            minim=mij;*/
    }
    //cout<<"dre"<<dr<<" ";
    else if(contor>k && st<mij && dr>mij){
        //if(k<=minim)
        //cout<<"minim";
            minim=mij;
        cauta(mij,dr);
    }
    //cout<<st<<" "<<dr<<" "<<contor;
}

int main()
{
    int i;
    ifstream f("transport.in");
    ofstream g("transport.out");
    f>>n>>k;
    int maxim=-1;
    for(i=0;i<n;i++){
        f>>a[i];
        suma=suma+a[i];
        if(a[i]>maxim)
            maxim=a[i];
    }

    int st=maxim;
    int dr=suma;
    int mij,nr;
    while(st<dr){
        mij=(st+dr)/2;
        nr=1;
        suma=0;
        for(i=0;i<n;i++)
            if(suma+a[i]<=mij){
                suma+=a[i];
            }
            else{
                suma=a[i];
                nr++;
            }
        if(k<nr)
            st=mij+1;
        else dr=mij;
    }
    //cauta(maxim, suma);
    g<<st;
    return 0;
}