#include <stdio.h> #include <stdbool.h> #include <algorithm> #include <vector> #include <map> using namespace std; bool bDebug = !true; class Point { public: Point(int v, int p) { val = v; pos = p; } int val; int pos; }; struct Query { int pos; int64_t ballance; bool bOn; }; bool operator<(const Query& l, const Query& r) { if (l.pos < r.pos ) return true; if (l.pos > r.pos ) return false; if (l.ballance < r.ballance) return true; if (l.ballance > r.ballance) return false; if (l.bOn < r.bOn ) return true; if (l.bOn > r.bOn ) return false; return false; } vector<Point> points; map<Query, int> m; __inline int cost(int pos) { return pos > 0 ? points[pos].pos - points[pos - 1].pos : 0; } int min_cost(int pos, int64_t ballance,bool on) { Query q = { pos,ballance,on }; if (m.find(q) != m.end()) return (int)m.find(q)->second; const int64_t v = points[pos].val + ballance; const int step_cost = (on ? cost(pos) : 0); if (pos == (int)points.size() - 1) { if (v<0) return 999999999; return step_cost; } if (v == 0) { return step_cost + min_cost(pos + 1, 0, false); } else if (v<0) { return step_cost + min_cost(pos + 1, v, true); } const int a = step_cost + min_cost(pos + 1, 0, false); const int b = step_cost + min_cost(pos + 1, v, true); const int res = a < b ? a : b; m[q] = res; return res; } int main() { int n; scanf("%d\n",&n); int64_t needs = 0; int64_t power = 0; points.reserve(5000); for (int i = 0; i < n; i++) { int v; scanf("%d",&v); if (v > 0) power += v; if (v < 0) needs += v; if (v != 0) { points.push_back(Point(v, i)); } } if (power < needs) { printf("-1\n"); return 0; } int res = min_cost(0, 0, false); if (res >= 999999999) { printf("-1\n"); return 0; } printf("%d\n", res); if (bDebug) while (true); return 0; }
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 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 | #include <stdio.h> #include <stdbool.h> #include <algorithm> #include <vector> #include <map> using namespace std; bool bDebug = !true; class Point { public: Point(int v, int p) { val = v; pos = p; } int val; int pos; }; struct Query { int pos; int64_t ballance; bool bOn; }; bool operator<(const Query& l, const Query& r) { if (l.pos < r.pos ) return true; if (l.pos > r.pos ) return false; if (l.ballance < r.ballance) return true; if (l.ballance > r.ballance) return false; if (l.bOn < r.bOn ) return true; if (l.bOn > r.bOn ) return false; return false; } vector<Point> points; map<Query, int> m; __inline int cost(int pos) { return pos > 0 ? points[pos].pos - points[pos - 1].pos : 0; } int min_cost(int pos, int64_t ballance,bool on) { Query q = { pos,ballance,on }; if (m.find(q) != m.end()) return (int)m.find(q)->second; const int64_t v = points[pos].val + ballance; const int step_cost = (on ? cost(pos) : 0); if (pos == (int)points.size() - 1) { if (v<0) return 999999999; return step_cost; } if (v == 0) { return step_cost + min_cost(pos + 1, 0, false); } else if (v<0) { return step_cost + min_cost(pos + 1, v, true); } const int a = step_cost + min_cost(pos + 1, 0, false); const int b = step_cost + min_cost(pos + 1, v, true); const int res = a < b ? a : b; m[q] = res; return res; } int main() { int n; scanf("%d\n",&n); int64_t needs = 0; int64_t power = 0; points.reserve(5000); for (int i = 0; i < n; i++) { int v; scanf("%d",&v); if (v > 0) power += v; if (v < 0) needs += v; if (v != 0) { points.push_back(Point(v, i)); } } if (power < needs) { printf("-1\n"); return 0; } int res = min_cost(0, 0, false); if (res >= 999999999) { printf("-1\n"); return 0; } printf("%d\n", res); if (bDebug) while (true); return 0; } |