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
| #include <bits/stdc++.h>
#define debug(x) cerr << #x << " = " << x << endl
using namespace std; typedef long long LL;
const int MAXN = 2E5 + 5; const int MOD = 1E9 + 7;
int n; int p[MAXN], s[MAXN]; LL sum = 0;
#define lid id << 1 #define rid id << 1 | 1
struct node { int l, r; int mi, lazy; } tr[MAXN << 2];
void update(int id) { tr[id].mi = min(tr[lid].mi, tr[rid].mi); }
void build(int id, int l, int r) { tr[id].l = l, tr[id].r = r; tr[id].lazy = 0; if (l == r) { tr[id].mi = s[l]; return; } int mid = l + r >> 1; build(lid, l, mid); build(rid, mid + 1, r); update(id); }
void pushdown(int id) { if (tr[id].lazy) { tr[lid].mi += tr[id].lazy; tr[rid].mi += tr[id].lazy; tr[lid].lazy += tr[id].lazy; tr[rid].lazy += tr[id].lazy; tr[id].lazy = 0; } }
void modify(int id, int l, int r, int v) { if (tr[id].l == l && tr[id].r == r) { tr[id].mi += v; tr[id].lazy += v; return; } pushdown(id); int mid = tr[id].l + tr[id].r >> 1; if (r <= mid) modify(lid, l, r, v); else if (l > mid) modify(rid, l, r, v); else modify(lid, l, mid, v), modify(rid, mid + 1, r, v); update(id); }
int qmin(int id, int l, int r) { if (tr[id].l == l && tr[id].r == r) return tr[id].mi; pushdown(id); int mid = tr[id].l + tr[id].r >> 1; if (r <= mid) return qmin(lid, l, r); else if (l > mid) return qmin(rid, l, r); else return min(qmin(lid, l, mid), qmin(rid, mid + 1, r)); }
signed main() { ios::sync_with_stdio(false), cin.tie(0); cin >> n; LL sum = 0; set<pair<int, int>> sp; for (int i = 1; i <= n; i++) { cin >> p[i]; sum -= p[i]; s[i] = s[i - 1] + 1; sp.insert({-p[i], -i}); } build(1, 1, n); while (sp.size()) { auto [val, id] = *sp.begin(); sp.erase(sp.begin()); val = -val; id = -id; int tmp = qmin(1, id, n); if (tmp >= 2) { sum += val * 2; modify(1, id, n, -2); } else if (tmp >= 1) { sum += val; modify(1, id, n, -1); } } cout << sum; return 0; }
|