Cod sursa(job #3345066)

Utilizator gaminggodBarbulescu Luca Traian gaminggod Data 7 martie 2026 20:16:20
Problema Transport Scor 20
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.01 kb
#include <bits/stdc++.h>

using namespace std;

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

#define cin fin
#define cout fout

const int Nmax=2e4;
int v[Nmax];

int main()
{
    int n,k;
    cin>>n>>k;
    int st=0,dr=0;
    for(int i=1;i<=n;i++)
    {
        cin>>v[i];
        dr+=v[i];
        st=max(st,v[i]);
    }
    int ans=0;
    while(st<=dr)
    {
        int m=(st+dr)/2;
        int nr=0,cnt=0,idx=0;
        for(int i=1;i<=n;i++)
        {
            nr+=v[i];
            if(nr==m)
            {
                nr=0;
                cnt++;

                idx=i;
            }
            else if(nr>m)
            {
                nr=v[i];
                cnt++;
                if(i==n)
                    cnt++;
                idx=i;
            }
        }
        if(cnt>k)
        {
            st=m+1;
        }
        else
        {
            dr=m-1;
            ans=m;
        }
    }
    cout<<ans;
    return 0;
}