【bzoj1367】左偏树

1367: [Baltic2004]sequence

Description

然而转化到递增序列，我们只需要将每个数读进来的之后减去它的下标就可以了

所以我们对于每一段已求好的序列，既要维护它的中位数，又要支持合并

根据这个性质我们又可以用可并堆了，堆顶元素表示该序列中的中位数

当堆的元素个数*2>序列长度+1的时候就可以弹出堆顶

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <string>
#define Rep(i, x, y) for (int i = x; i <= y; i ++)
#define RepE(i, x) for (int i = pos[x]; i; i = g[i].nex)
#define Dwn(i, x, y) for (int i = x; i >= y; i --)
#define u t[x]
#define v t[y]
#define Lc t[ u.lc ]
#define Rc t[ u.rc ]
using namespace std;
typedef long long LL;
const int N = 1000005;
struct arr { int lc, rc, vl, d; } t
;
int n, ro
, l
, s
, a
, m, l1
, r1
;
LL ans;
int Merge(int x, int y) {
if (!x || !y) return x + y;
if (u.vl < v.vl) swap(x, y);
u.rc = Merge(u.rc, y);
if (Rc.d > Lc.d) swap(u.rc, u.lc);
u.d = Rc.d + 1;
return x;
}
int Del(int x) {
return Merge(u.lc, u.rc);
}
int main()
{
scanf ("%d", &n);
Rep(i, 1, n) {
scanf ("%d", &a[i]), a[i] -= i;
}
Rep(i, 1, n) {
t[ro[++ m] = i].vl = a[i], l[m] = s[m] = 1, l1[m] = r1[m] = i;
while (m > 1 && t[ ro[m-1] ].vl > t[ ro[m] ].vl) {
l[m-1] = l[m] + l[m-1], s[m-1] += s[m];
ro[m - 1] = Merge(ro[m], ro[m-1]);
m --, r1[m] = i;
while (l[m] > (s[m] + 1) / 2) ro[m] = Del(ro[m]), l[m] --;
}
}
Rep(i, 1, m) {
int k = t[ ro[i] ].vl;
Rep(j, l1[i], r1[i]) {
ans += abs(a[j] - k);
}
}
printf("%lld\n", ans);

return 0;
}