洛谷P2234[营业额统计]

戳这里看原题

---

这道题目一看就是要求在每个数前面的一段前缀中的前驱和后继。

很直白地就会想到二叉平衡树。

对于每一天的最小波动值只需要在二叉平衡树中插入节点的时候记录前驱和后继就可以很方便地算出来了。

所以总的来说，这道题就是一道模板题

先贴上treap的板子吧。。

---

//treap
#include <bits/stdc++.h>

using namespace std;

const int N = 32768;

struct treap {
    int fa;
    int s[2];
    int wei;
    int key;
    
    friend void check_print(const treap a) {
        printf("!%d %d %d %d %d
", a.fa, a.s[0], a.s[1], a.wei, a.key);
    }
}tp[N + 1];

int n;
int root, tot;
int ans;

void rotate(int x) {
    int y = tp[x].fa;
    int z = tp[y].fa;
    int c = tp[y].s[1] == x;
    tp[x].fa = z;
    if(z) tp[z].s[tp[z].s[1] == y] = x;
    tp[tp[x].s[c ^ 1]].fa = y;
    tp[y].s[c] = tp[x].s[c ^ 1];
    tp[x].s[c ^ 1] = y;
    tp[y].fa = x;
    if(root == y) root = x;
}

void set_up(int k, int f, int x) {
    tp[k].key = x;
    tp[k].wei = rand();
    tp[k].fa = f;
    tp[f].s[x > tp[f].key] = k;
}

void up(int k) {
    int f = tp[k].fa;
    while(f && tp[f].wei > tp[k].wei) {
        rotate(k);
        f = tp[k].fa;
    }
}

int insert(int k, int x, int b, int s) {
    if(tp[k].key == x) return 0;
    if(tp[k].key > x) s = min(s, tp[k].key);
    if(tp[k].key < x) b = max(b, tp[k].key);
    if(!k || !tp[k].s[x > tp[k].key]) {
        set_up(++tot, k, x);
        up(tot);
        if(!k) root = tot;
        return min(abs(b - x), abs(s - x));
    }
    return insert(tp[k].s[tp[k].key < x], x, b, s);
}

int main() {
    srand(time(NULL));
    int a;
    scanf("%d%d", &n, &a);
    ans += a;
    insert(root, a, -2e9, 2e9);
    for(int i = 1; i < n; i++) {
        scanf("%d", &a);
        ans += insert(root, a, -2e9, 2e9);
    }
    printf("%d
", ans);
    return 0;
}

---

但是C++的STL库中有一个神奇的东西——set。

它完美的（带了巨大常数）实现了高难度的红黑树。

既然有这样的好东西，为何不用呢？

---

#include <bits/stdc++.h>

using namespace std;

const int N = 100000;

int n;
int a;
int ans;
set<int>s;

int main() {
    scanf("%d", &n);
    scanf("%d", &a);
    s.insert(2e9);
    s.insert(-2e9);
    s.insert(2e9 + 1);
    s.insert(-2e9 - 1);
    s.insert(a);
    ans += a;
    for(int i = 1; i < n; i++) {
        int k1 = 1e9, k2 = 1e9, k3 = 1e9, a;
        scanf("%d", &a);
        set<int>::iterator it = s.upper_bound(a);
        k1 = abs(a - *it);
        k2 = abs(a - *it), it--;
        k3 = abs(a - *it), it++;
        ans += min(k1, min(k2, k3));
        s.insert(a);
    }
    printf("%d
", ans);
    return 0;
}

---

至于set的用法，这里给大家推荐一个个人认为归纳的比较好的博客，可以去借鉴一下。

戳这里