Cod sursa(job #2642373)

Utilizator cyg_vladioanBirsan Vlad cyg_vladioan Data 14 august 2020 23:09:42
Problema Ferma Scor 10
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.31 kb
#include <cstdio>
#include <algorithm>
using namespace std;
const int NMAX = 10000;
int v[NMAX + 5] , sp[NMAX + 5] , sn[NMAX + 5] , vn[NMAX + 5];
int main()
{
    freopen("ferma.in" , "r" , stdin);
    freopen("ferma.out" , "w" , stdout);
    int n , k , i , cntp , np , nn;
    long long s = 0;
    scanf("%d%d" , &n , &k);
    for(i = 1 ; i <= n ; i ++)
        scanf("%d" , &v[i]);
    cntp = 0;
    np = nn = 1;
    for(i = 1 ; i <= n ; i ++)
    {
        if(v[i] >= 0)
        {
            sp[np] += v[i];
            cntp ++;
        }
        else if(i > 1 && v[i] < 0 && v[i - 1] >= 0)
            np ++;
    }
    if(v[n] >= 0 && v[1] >= 0)
    {
        sp[1] += sp[np];
        np --;
    }
    for(i = 1 ; i <= n ; i ++)
    {
        if(v[i] < 0)
        {
            sn[nn] += v[i];
            vn[++ vn[0]] = v[i];
        }
        else if(i > 1 && v[i] >= 0 && v[i - 1] < 0)
            nn --;
    }
    if(v[n] < 0 && v[1] < 0)
    {
        sn[1] += sn[nn];
        nn --;
    }
    if(np <= k)
        for(i = 1 ; i <= np ; i ++)
            s += sp[i];
    if(np < k && cntp < k)
    {
        sort(vn + 1 , vn + vn[0] + 1);
        for(i = vn[0] ; i >= vn[0] - (k - cntp) + 1 ; i --)
            s += vn[i];
    }
    printf("%lld\n" , s);
    return 0;
}