#include <cstdio> #include <iostream> #include <algorithm> #include <cstring> #include <vector> #include <set> #include <map> #include <array> #include <random> #include <cmath> #include <list> #include <ctime> #include <sstream> #include <queue> #include <climits> #include <stack> #include <random> #include <bitset> #include <numeric> #include <cassert> using namespace std; typedef vector<int> vi; typedef pair<int,int> pii; typedef long long ll; #define rep(x, b, e) for(int x=(b); x<(e); ++x) #define trav(a, x) for(auto& a : x) #define ford(x, b, e) for(int x=((int)(b))-1; x>=(e); --x) #define all(c) c.begin(),c.end() #define sz(x) ((int)((x).size())) #define pb push_back #define st first #define nd second #define mp(x,y) make_pair(x,y) typedef short int sint; #define imie(...) " [" << #__VA_ARGS__ ": " << (__VA_ARGS__) << "] " const int N = 1 << 20; int P; int t[N]; int getMin(int p, int q, int pocz = 1, int kon = P, int ind = 1) { if (p == pocz && q == kon) return t[ind]; int sr = (pocz + kon) / 2; if (q <= sr) return getMin(p, q, pocz, sr, 2*ind); else if (p > sr) return getMin(p, q, sr + 1, kon, 2*ind + 1); return min(getMin(p, sr, pocz, sr, 2*ind), getMin(sr+1, q, sr+1, kon, 2*ind + 1)); } void setuj(int pos, int val) { pos = pos + P - 1; t[pos] = val; while (pos/2) { pos /= 2; t[pos] = min(t[2*pos], t[2*pos + 1]); } } int main() { ios_base::sync_with_stdio(false); cin.tie(0); int n; scanf("%d", &n); ll s = 0; vector<pair<ll, int>> events; events.pb({s, 1}); rep(i, 0, n) { int x; scanf("%d", &x); s += x; events.pb({s, i + 2}); } ++n; vi f(n + 1, INT_MAX); P = 1; while (P < n) P <<= 1; // printf("n: %d, P: %d\n", n, P); rep(i, 1, 2 * P) { t[i] = INT_MAX; } sort(all(events)); trav(ev, events) { // rozwazam ev.nd int best; if (ev.nd == 1) { best = 0; } else { best = getMin(1, ev.nd); } if (best != INT_MAX) { best = best + ev.nd - 1; } f[ev.nd] = best; if (f[ev.nd] != INT_MAX) { setuj(ev.nd, f[ev.nd] - ev.nd); } } printf("%d\n", f[n] == INT_MAX ? -1 : f[n]); }
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 | #include <cstdio> #include <iostream> #include <algorithm> #include <cstring> #include <vector> #include <set> #include <map> #include <array> #include <random> #include <cmath> #include <list> #include <ctime> #include <sstream> #include <queue> #include <climits> #include <stack> #include <random> #include <bitset> #include <numeric> #include <cassert> using namespace std; typedef vector<int> vi; typedef pair<int,int> pii; typedef long long ll; #define rep(x, b, e) for(int x=(b); x<(e); ++x) #define trav(a, x) for(auto& a : x) #define ford(x, b, e) for(int x=((int)(b))-1; x>=(e); --x) #define all(c) c.begin(),c.end() #define sz(x) ((int)((x).size())) #define pb push_back #define st first #define nd second #define mp(x,y) make_pair(x,y) typedef short int sint; #define imie(...) " [" << #__VA_ARGS__ ": " << (__VA_ARGS__) << "] " const int N = 1 << 20; int P; int t[N]; int getMin(int p, int q, int pocz = 1, int kon = P, int ind = 1) { if (p == pocz && q == kon) return t[ind]; int sr = (pocz + kon) / 2; if (q <= sr) return getMin(p, q, pocz, sr, 2*ind); else if (p > sr) return getMin(p, q, sr + 1, kon, 2*ind + 1); return min(getMin(p, sr, pocz, sr, 2*ind), getMin(sr+1, q, sr+1, kon, 2*ind + 1)); } void setuj(int pos, int val) { pos = pos + P - 1; t[pos] = val; while (pos/2) { pos /= 2; t[pos] = min(t[2*pos], t[2*pos + 1]); } } int main() { ios_base::sync_with_stdio(false); cin.tie(0); int n; scanf("%d", &n); ll s = 0; vector<pair<ll, int>> events; events.pb({s, 1}); rep(i, 0, n) { int x; scanf("%d", &x); s += x; events.pb({s, i + 2}); } ++n; vi f(n + 1, INT_MAX); P = 1; while (P < n) P <<= 1; // printf("n: %d, P: %d\n", n, P); rep(i, 1, 2 * P) { t[i] = INT_MAX; } sort(all(events)); trav(ev, events) { // rozwazam ev.nd int best; if (ev.nd == 1) { best = 0; } else { best = getMin(1, ev.nd); } if (best != INT_MAX) { best = best + ev.nd - 1; } f[ev.nd] = best; if (f[ev.nd] != INT_MAX) { setuj(ev.nd, f[ev.nd] - ev.nd); } } printf("%d\n", f[n] == INT_MAX ? -1 : f[n]); } |