#include <cstdio> #include <vector> #include <map> using namespace std; #define FOR(i,a,b) for(int (i)=(int)(a); (i)!=(int)(b); ++(i)) int n; vector<long long int> A; map<long long int, int> M; long long int base_extra; int base_length; typedef map<long long int, int>::iterator Iter; int main() { // Read input scanf("%d", &n); A.resize(n); int a; long long int sum = 0; FOR(i,0,n) { scanf("%d", &a); A[i] = (long long int)(a); sum += A[i]; } if (sum < 0) { printf("-1\n"); return 0; } base_length = 0; base_extra = 0; M[0] = 0; FOR(i,0,n) { int best_real_length = -1; Iter best_it = M.lower_bound(base_extra); if (best_it != M.end()) best_real_length = best_it->second - base_length; base_extra -= A[i]; base_length -= 1; Iter inserted_it = M.end(); if (best_real_length != -1) { long long int repr_extra = A[i] + base_extra; int repr_length = best_real_length + base_length; Iter it = M.find(repr_extra); if (it != M.end()) { if (it->second > repr_length) { it->second = repr_length; inserted_it = it; } } else { M[repr_extra] = repr_length; inserted_it = M.find(repr_extra); } } if (inserted_it != M.end()) { Iter it = inserted_it; ++it; if (it != M.end() && it->second <= inserted_it->second) { M.erase(inserted_it); } else { while (inserted_it != M.begin()) { it = inserted_it; --it; if (it->second >= inserted_it->second) M.erase(it); else break; } } } } Iter best_it = M.lower_bound(base_extra); if (best_it == M.end()) printf("-1\n"); printf("%d\n", best_it->second - base_length); }
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 | #include <cstdio> #include <vector> #include <map> using namespace std; #define FOR(i,a,b) for(int (i)=(int)(a); (i)!=(int)(b); ++(i)) int n; vector<long long int> A; map<long long int, int> M; long long int base_extra; int base_length; typedef map<long long int, int>::iterator Iter; int main() { // Read input scanf("%d", &n); A.resize(n); int a; long long int sum = 0; FOR(i,0,n) { scanf("%d", &a); A[i] = (long long int)(a); sum += A[i]; } if (sum < 0) { printf("-1\n"); return 0; } base_length = 0; base_extra = 0; M[0] = 0; FOR(i,0,n) { int best_real_length = -1; Iter best_it = M.lower_bound(base_extra); if (best_it != M.end()) best_real_length = best_it->second - base_length; base_extra -= A[i]; base_length -= 1; Iter inserted_it = M.end(); if (best_real_length != -1) { long long int repr_extra = A[i] + base_extra; int repr_length = best_real_length + base_length; Iter it = M.find(repr_extra); if (it != M.end()) { if (it->second > repr_length) { it->second = repr_length; inserted_it = it; } } else { M[repr_extra] = repr_length; inserted_it = M.find(repr_extra); } } if (inserted_it != M.end()) { Iter it = inserted_it; ++it; if (it != M.end() && it->second <= inserted_it->second) { M.erase(inserted_it); } else { while (inserted_it != M.begin()) { it = inserted_it; --it; if (it->second >= inserted_it->second) M.erase(it); else break; } } } } Iter best_it = M.lower_bound(base_extra); if (best_it == M.end()) printf("-1\n"); printf("%d\n", best_it->second - base_length); } |