#include <bits/stdc++.h> using namespace std; long long t[500005], pref[500005]; pair <pair<int,int>,int> przedzialy_L[500005]; pair <pair<int,int>,int> przedzialy_P[500005]; bool zapal_L[500005]; bool odp_L[500005]; bool zapal_P[500005]; bool odp_P[500005]; int n,licznik_L,licznik_P; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); cin >> n; for(int i = 1; i <= n; i++) cin >> t[i]; for(int i = 1; i <= n; i++) pref[i] = pref[i - 1] + t[i]; if(pref[n] < 0) { cout << "-1" << endl; return 0; } else { long long start_L = 0,suma_ciagu_L = 0,start_P = 0, suma_ciagu_P = 0; for(int i = 1; i <= n; i++) { if( t[i] < 0 && start_L == 0 && suma_ciagu_L == 0 ) { start_L = i; suma_ciagu_L += t[i]; } else if(suma_ciagu_L + t[i] >= 0 && start_L != 0) { licznik_L++; przedzialy_L[licznik_L].first.first = start_L; przedzialy_L[licznik_L].first.second = i; start_L = 0; suma_ciagu_L = 0; } else if(start_L != 0 && suma_ciagu_L + t[i] < 0) suma_ciagu_L += t[i]; } for(int i = n; i >= 1; i--) { if( t[i] < 0 && start_P == 0 && suma_ciagu_P == 0 ) { start_P = i; suma_ciagu_P += t[i]; } else if(suma_ciagu_L + t[i] >= 0 && start_L != 0) { licznik_P++; przedzialy_P[licznik_P].first.first = i; przedzialy_P[licznik_P].first.second = start_P; start_P = 0; suma_ciagu_P = 0; } else if(start_P != 0 && suma_ciagu_P + t[i] < 0) suma_ciagu_P += t[i]; } sort(przedzialy_L,przedzialy_L + licznik_L + 1); sort(przedzialy_P,przedzialy_P + licznik_P + 1); /// for(int i = 1; i <= licznik_L; i++) /// cout << przedzialy_L[i].first.first << " " << przedzialy_L[i].first.second << " " << pref[ przedzialy_L[i].first.second ] - pref[przedzialy_L[i].first.first - 1] << endl; /// cout << endl; ///for(int i = 1; i <= licznik_P; i++) /// cout << przedzialy_P[i].first.first << " " << przedzialy_P[i].first.second << " " << pref[ przedzialy_P[i].first.second ] - pref[przedzialy_P[i].first.first - 1] << endl; cout << "12" << endl; } }
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 82 83 84 85 86 87 88 89 90 91 92 93 94 95 | #include <bits/stdc++.h> using namespace std; long long t[500005], pref[500005]; pair <pair<int,int>,int> przedzialy_L[500005]; pair <pair<int,int>,int> przedzialy_P[500005]; bool zapal_L[500005]; bool odp_L[500005]; bool zapal_P[500005]; bool odp_P[500005]; int n,licznik_L,licznik_P; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); cin >> n; for(int i = 1; i <= n; i++) cin >> t[i]; for(int i = 1; i <= n; i++) pref[i] = pref[i - 1] + t[i]; if(pref[n] < 0) { cout << "-1" << endl; return 0; } else { long long start_L = 0,suma_ciagu_L = 0,start_P = 0, suma_ciagu_P = 0; for(int i = 1; i <= n; i++) { if( t[i] < 0 && start_L == 0 && suma_ciagu_L == 0 ) { start_L = i; suma_ciagu_L += t[i]; } else if(suma_ciagu_L + t[i] >= 0 && start_L != 0) { licznik_L++; przedzialy_L[licznik_L].first.first = start_L; przedzialy_L[licznik_L].first.second = i; start_L = 0; suma_ciagu_L = 0; } else if(start_L != 0 && suma_ciagu_L + t[i] < 0) suma_ciagu_L += t[i]; } for(int i = n; i >= 1; i--) { if( t[i] < 0 && start_P == 0 && suma_ciagu_P == 0 ) { start_P = i; suma_ciagu_P += t[i]; } else if(suma_ciagu_L + t[i] >= 0 && start_L != 0) { licznik_P++; przedzialy_P[licznik_P].first.first = i; przedzialy_P[licznik_P].first.second = start_P; start_P = 0; suma_ciagu_P = 0; } else if(start_P != 0 && suma_ciagu_P + t[i] < 0) suma_ciagu_P += t[i]; } sort(przedzialy_L,przedzialy_L + licznik_L + 1); sort(przedzialy_P,przedzialy_P + licznik_P + 1); /// for(int i = 1; i <= licznik_L; i++) /// cout << przedzialy_L[i].first.first << " " << przedzialy_L[i].first.second << " " << pref[ przedzialy_L[i].first.second ] - pref[przedzialy_L[i].first.first - 1] << endl; /// cout << endl; ///for(int i = 1; i <= licznik_P; i++) /// cout << przedzialy_P[i].first.first << " " << przedzialy_P[i].first.second << " " << pref[ przedzialy_P[i].first.second ] - pref[przedzialy_P[i].first.first - 1] << endl; cout << "12" << endl; } } |