1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
#include <iostream>
#include <queue>
using namespace std;

int n;
vector <bool> odwiedzone;
queue <int> kolejka;
vector <int> graf;
vector <int> road_length;
int wynik=0;
int licznik=0;
int poczatkowy_wierzcholek;

void BFS(int wierzcholek)
{
    odwiedzone[wierzcholek]=1;

    if (graf[wierzcholek]>0)
    {
        licznik+=graf[wierzcholek];

        if (licznik>=0)
        {
            wynik+=abs(road_length[wierzcholek]-road_length[poczatkowy_wierzcholek]);
        }
    }
    //cout<<"xd";
    if(licznik<0)
    {
        kolejka.push(wierzcholek);
        if(wierzcholek<n-1 && odwiedzone[wierzcholek+1]==0)
            kolejka.push(wierzcholek+1);

        if(wierzcholek>0 && odwiedzone[wierzcholek-1]==0)
            kolejka.push(wierzcholek-1);
        kolejka.pop();

        if(!kolejka.empty())
            BFS(kolejka.front());
    }

}

int main()
{
    int suma=0;
    cin>>n;
    odwiedzone.resize(n,0);
    for (int i=0; i<n; i++)
    {
        int a;
        cin>>a;
        suma+=a;
        road_length.push_back(i+1);
        graf.push_back(a);

    }

    if (suma<0)
    {
        cout<<"-1";
        return 0;
    }

    for (int i=0; i<n; i++)
    {
        if (graf[i]<0)
        {
            licznik=graf[i];
            poczatkowy_wierzcholek=i;
            BFS(i);

            odwiedzone.clear();
            odwiedzone.resize(n,0);
        }

    }

cout<<wynik;
    return 0;
}