#include <bits/stdc++.h>
using namespace std;
#define inf 1000000000000000000ll
#define int long long
#define N 500005
#define T (1 << 20)
int n;
int a[N];
int dp[N];
int pf[N];
vector<int> b;
int t[T << 1];
int p[T << 1];
void push(int u) {
t[u] += p[u];
if(u < T) {
p[u << 1 | 0] += p[u];
p[u << 1 | 1] += p[u];
}
p[u] = 0;
}
void add(int l, int r, int v, int x = 0, int y = T, int u = 1) {
assert(0 <= l && l < r && r <= T);
push(u);
if(l >= y || r <= x || v == 0) return;
if(l <= x && y <= r) {
p[u] += v;
push(u);
return;
}
int m = (x + y) >> 1;
add(l, r, v, x, m, u << 1 | 0);
add(l, r, v, m, y, u << 1 | 1);
t[u] = min(t[u << 1 | 0], t[u << 1 | 1]);
}
int query(int l, int r, int x = 0, int y = T, int u = 1) {
assert(0 <= l && l < r && r <= T);
push(u);
if(l >= y || r <= x) return LLONG_MAX;
if(l <= x && y <= r) return t[u];
int m = (x + y) >> 1;
return min(query(l, r, x, m, u << 1 | 0),
query(l, r, m, y, u << 1 | 1));
}
void update(int pos, int val) {
add(pos, pos + 1, min(0ll, val - query(pos, pos + 1)));
}
int32_t main() {
for(int i = 0; i < (T << 1); i++) {
t[i] = inf;
}
scanf("%lld", &n);
for(int i = 1; i <= n; i++) {
scanf("%lld", a + i);
pf[i] = pf[i - 1] + a[i];
b.push_back(pf[i]);
}
b.push_back(0);
sort(b.begin(), b.end());
b.erase(unique(b.begin(), b.end()), b.end());
auto get_id = [&](int i) {
return (int)(lower_bound(b.begin(), b.end(), i) - b.begin());
};
update(get_id(0), 0);
for(int i = 1; i <= n; i++) {
if(a[i] >= 0) dp[i] = dp[i - 1];
else dp[i] = inf;
dp[i] = min(dp[i], query(0, get_id(pf[i]) + 1));
update(get_id(pf[i]), dp[i] - 1);
add(0, T, 1);
}
// for(int i = 1; i <= n; i++) cerr << dp[i] << ' ';
// cerr << '\n';
if(dp[n] >= inf) puts("-1");
else printf("%lld\n", dp[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 | #include <bits/stdc++.h> using namespace std; #define inf 1000000000000000000ll #define int long long #define N 500005 #define T (1 << 20) int n; int a[N]; int dp[N]; int pf[N]; vector<int> b; int t[T << 1]; int p[T << 1]; void push(int u) { t[u] += p[u]; if(u < T) { p[u << 1 | 0] += p[u]; p[u << 1 | 1] += p[u]; } p[u] = 0; } void add(int l, int r, int v, int x = 0, int y = T, int u = 1) { assert(0 <= l && l < r && r <= T); push(u); if(l >= y || r <= x || v == 0) return; if(l <= x && y <= r) { p[u] += v; push(u); return; } int m = (x + y) >> 1; add(l, r, v, x, m, u << 1 | 0); add(l, r, v, m, y, u << 1 | 1); t[u] = min(t[u << 1 | 0], t[u << 1 | 1]); } int query(int l, int r, int x = 0, int y = T, int u = 1) { assert(0 <= l && l < r && r <= T); push(u); if(l >= y || r <= x) return LLONG_MAX; if(l <= x && y <= r) return t[u]; int m = (x + y) >> 1; return min(query(l, r, x, m, u << 1 | 0), query(l, r, m, y, u << 1 | 1)); } void update(int pos, int val) { add(pos, pos + 1, min(0ll, val - query(pos, pos + 1))); } int32_t main() { for(int i = 0; i < (T << 1); i++) { t[i] = inf; } scanf("%lld", &n); for(int i = 1; i <= n; i++) { scanf("%lld", a + i); pf[i] = pf[i - 1] + a[i]; b.push_back(pf[i]); } b.push_back(0); sort(b.begin(), b.end()); b.erase(unique(b.begin(), b.end()), b.end()); auto get_id = [&](int i) { return (int)(lower_bound(b.begin(), b.end(), i) - b.begin()); }; update(get_id(0), 0); for(int i = 1; i <= n; i++) { if(a[i] >= 0) dp[i] = dp[i - 1]; else dp[i] = inf; dp[i] = min(dp[i], query(0, get_id(pf[i]) + 1)); update(get_id(pf[i]), dp[i] - 1); add(0, T, 1); } // for(int i = 1; i <= n; i++) cerr << dp[i] << ' '; // cerr << '\n'; if(dp[n] >= inf) puts("-1"); else printf("%lld\n", dp[n]); } |
English