#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll val[500009];
ll pref[500009];
unordered_map<ll, ll> calc;
inline bool ok(int a, int b) {
return (pref[b+1] - pref[a]) >= 0;
}
inline int cost(int a, int b) {
return b - a;
}
ll solve(ll a, ll b) {
if(not ok(a, b))
return -1;
ll res = cost(a, b);
ll last = a;
for(ll i = a + 1; i <= b; i++) {
if(val[i] == 0)
continue;
if(ok(a, last) and ok(i, b)) {
auto it = calc.find(1000000*i+b);
if(it == calc.end())
res = min(res, cost(a, last) + solve(i, b));
else
res = min(res, cost(a, last) + (*it).second);
}
last = i;
}
calc[1000000*a+b] = res;
return res;
}
int main() {
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
int n;
cin >> n;
for(int i = 1; i <= n; i++) {
cin >> val[i];
pref[i+1] = pref[i] + val[i];
}
int a = 1, b = n;
while(val[a] == 0 and a <= n)
a++;
if(a > n)
cout << 0 << "\n";
else {
while(val[b] == 0)
b--;
cout << solve(a, b) << "\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 | #include<bits/stdc++.h> #define ll long long using namespace std; ll val[500009]; ll pref[500009]; unordered_map<ll, ll> calc; inline bool ok(int a, int b) { return (pref[b+1] - pref[a]) >= 0; } inline int cost(int a, int b) { return b - a; } ll solve(ll a, ll b) { if(not ok(a, b)) return -1; ll res = cost(a, b); ll last = a; for(ll i = a + 1; i <= b; i++) { if(val[i] == 0) continue; if(ok(a, last) and ok(i, b)) { auto it = calc.find(1000000*i+b); if(it == calc.end()) res = min(res, cost(a, last) + solve(i, b)); else res = min(res, cost(a, last) + (*it).second); } last = i; } calc[1000000*a+b] = res; return res; } int main() { ios::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n; cin >> n; for(int i = 1; i <= n; i++) { cin >> val[i]; pref[i+1] = pref[i] + val[i]; } int a = 1, b = n; while(val[a] == 0 and a <= n) a++; if(a > n) cout << 0 << "\n"; else { while(val[b] == 0) b--; cout << solve(a, b) << "\n"; } return 0; } |
English