#include<bits/stdc++.h>
using namespace std;
using lld = long long;
const lld INF = 1e16;
struct ST{
vector<lld> tree;
int base;
ST(int n=(1<<20)): base(1) {
while(base < n)
base <<= 1;
tree.assign(2*base,INF);
}
void insert(int x, lld val){
x += base;
tree[x] = val;
while(x > 1){
x /= 2;
tree[x] = min(tree[2*x],tree[2*x+1]);
}
}
lld query(int b, int e){
b += base, e += base;
lld val = min(tree[b], tree[e]);
while(b/2 != e/2){
if(!(b&1))
val = min(val, tree[b+1]);
if(e&1)
val = min(val, tree[e-1]);
b /= 2, e /= 2;
}
return val;
}
};
int main(void){
ios_base::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
vector<lld> t(n);
for(lld& i : t)
cin >> i;
vector<lld> p(n+1,0);
for(int i=0;i<n;i++)
p[i+1] = p[i]+t[i];
if(p[n] < 0){
cout << "-1\n";
return 0;
}
vector<pair<lld,int> > ord(n+1);
for(int i=0;i<=n;i++)
ord[i] = {p[i], i};
sort(ord.begin(), ord.end());
ST dp(n+1);
for(auto [pref, i] : ord){
if(i == 0){
dp.insert(i,-1);
continue;
}
lld val = dp.query(0,i-1)+i;
dp.insert(i,val-(i+1));
}
cout << dp.query(n,n)+(n+1) << "\n";
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 | #include<bits/stdc++.h> using namespace std; using lld = long long; const lld INF = 1e16; struct ST{ vector<lld> tree; int base; ST(int n=(1<<20)): base(1) { while(base < n) base <<= 1; tree.assign(2*base,INF); } void insert(int x, lld val){ x += base; tree[x] = val; while(x > 1){ x /= 2; tree[x] = min(tree[2*x],tree[2*x+1]); } } lld query(int b, int e){ b += base, e += base; lld val = min(tree[b], tree[e]); while(b/2 != e/2){ if(!(b&1)) val = min(val, tree[b+1]); if(e&1) val = min(val, tree[e-1]); b /= 2, e /= 2; } return val; } }; int main(void){ ios_base::sync_with_stdio(false); cin.tie(nullptr); int n; cin >> n; vector<lld> t(n); for(lld& i : t) cin >> i; vector<lld> p(n+1,0); for(int i=0;i<n;i++) p[i+1] = p[i]+t[i]; if(p[n] < 0){ cout << "-1\n"; return 0; } vector<pair<lld,int> > ord(n+1); for(int i=0;i<=n;i++) ord[i] = {p[i], i}; sort(ord.begin(), ord.end()); ST dp(n+1); for(auto [pref, i] : ord){ if(i == 0){ dp.insert(i,-1); continue; } lld val = dp.query(0,i-1)+i; dp.insert(i,val-(i+1)); } cout << dp.query(n,n)+(n+1) << "\n"; return 0; } |
English