排序机械臂

Posted 2020-11-24 hs-black

FHQ Treap解法

这道题当然用好写的fhq解决啦(其实是不会splay)

一开始, 感觉无法同时权值分裂又排名分裂

所以我按排名分裂, 维护子树最小值, 设计一个类似求第k大的函数, 找出区间最小值的位置

详见代码, 还是很好懂的(除get_rk函数, 其他部分和文艺平衡树一样)

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<ctime>

using namespace std;

template <typename T>
void read(T &x) {
    x = 0; bool f = 0;
    char c = getchar();
    for (;!isdigit(c);c=getchar()) if (c=='-') f = 1;
    for (;isdigit(c);c=getchar()) x=(x<<3)+(x<<1)+(c^48);
    if (f) x=-x;
}

const int N = 100050;
int tag[N], siz[N];
int son[N][2], rnd[N];
int val[N], mn[N];

struct node {
    int pos, num;
    bool operator < (const node &i) const {
        if (num != i.num) return num < i.num;
        return pos < i.pos;
    }
}a[N];
int n;

int rt, x, y, z;
int tot;
int build(int x) {
    val[++tot] = x, mn[tot] = x, siz[tot] = 1;
    rnd[tot] = rand(); return tot;
}

int Mn(int x,int y) {return x < y ? x : y;}

void update(int x) {
    siz[x] = siz[son[x][0]] + siz[son[x][1]] + 1;
    mn[x] = val[x]; 
    if (son[x][0]) mn[x] = Mn(mn[son[x][0]], mn[x]);
    if (son[x][1]) mn[x] = Mn(mn[son[x][1]], mn[x]);
}

void spread(int x) {
    if (!tag[x]) return;
    swap(son[x][0], son[x][1]);
    if (son[x][0]) tag[son[x][0]] ^= 1;
    if (son[x][1]) tag[son[x][1]] ^= 1;
    tag[x] = 0;
}

int merge(int x,int y) {
    if (!x || !y) return x | y;
    if (rnd[x] < rnd[y]) {
        spread(x);
        son[x][1] = merge(son[x][1], y);
        update(x); return x;
    }
    spread(y);
    son[y][0] = merge(x, son[y][0]);
    update(y); return y;
}

void split(int now,int k,int &x,int &y) {
    if (!now) {
        x = y = 0; return;
    }
    spread(now);
    if (siz[son[now][0]] < k) {
        x = now;
        split(son[x][1], k - siz[son[x][0]] - 1, son[x][1], y);
    }
    else y = now, split(son[y][0], k, x, son[y][0]);
    update(now);
}

//找出最小值的排名
int get_rk(int x) {
    int k = 1; //初始排名
    while (1) {
        spread(x);
        if (son[x][0] && mn[son[x][0]] == mn[x]) x = son[x][0]; //在左子树
        else if (son[x][1] && mn[son[x][1]] == mn[x]) k += siz[son[x][0]] + 1, x = son[x][1]; 
        //在右子树
        else return k + siz[son[x][0]]; //在当前节点
    }
}

int v[N];
int main() {
    read(n);
    for (int i = 1;i <= n; i++) {
        read(a[i].num); a[i].pos = i;
    }
    sort(a + 1, a + n + 1);
    for (int i = 1;i <= n; i++)
    v[a[i].pos] = i;
    for (int i = 1;i <= n; i++) 
        rt = merge(rt, build(v[i]));
    for (int i = 1;i <= n; i++) {
        int k = get_rk(rt);
        split(rt, k, x, y);
        split(x, k-1, x, z);
        tag[x] ^= 1;
        rt = merge(x, y);
        printf ("%d ", k + i - 1);
    }
    return 0;
}
/*
7
1 8 6 5 3 5 2

*/