luogu-P1110(treap)

题目描述
Q的妈妈是一个出纳,经常需要做一些统计报表的工作。今天是妈妈的生日,小Q希望可以帮妈妈分担一些工作,作为她的生日礼物之一。

经过仔细观察,小Q发现统计一张报表实际上是维护一个非负整数数列,并且进行一些查询操作。

在最开始的时候,有一个长度为N的整数序列,并且有以下三种操作:

INSERT i k:在原数列的第i个元素后面添加一个新元素kk;如果原数列的第ii个元素已经添加了若干元素,则添加在这些元素的最后(见下面的例子)
MIN_GAP:查询相邻两个元素的之间差值(绝对值)的最小值
MIN_SORT_GAP:查询所有元素中最接近的两个元素的差值(绝对值)
例如一开始的序列为5, 3, 15,3,1。

执行操作INSERT 2 9将得到:5, 3, 9, 15,3,9,1,此时MIN_GAP为22,MIN_SORT_GAP为22。

再执行操作INSERT 2 6将得到:5, 3, 9, 6, 15,3,9,6,1

注意这个时候原序列的第22个元素后面已经添加了一个99,此时添加的66应加在99的后面。这个时候MIN_GAP为22,MIN_SORT_GAP为11。

于是小Q写了一个程序,使得程序可以自动完成这些操作,但是他发现对于一些大的报表他的程序运行得很慢,你能帮助他改进程序么?

输入格式
第一行包含两个整数NN,MM,分别表示原数列的长度以及操作的次数。

第二行为NN个整数,为初始序列。

接下来的MM行每行一个操作,即INSERT i kMIN_GAPMIN_SORT_GAP 中的一种(无多余空格或者空行)。

输出格式
对于每一个MIN_GAPMIN_SORT_GAP命令,输出一行答案即可。

输入输出样例
输入

3 5
5 3 1
INSERT 2 9
MIN_SORT_GAP
INSERT 2 6
MIN_GAP
MIN_SORT_GAP

输出

2
2
1

说明/提示
对于30%的数据,N ≤ 1000N≤1000 ,M ≤ 5000,M≤5000
对于100%的数据,N ,M ≤500000
对于所有的数据,序列内的整数不超过 5 × 1 0 8 5 × 10^8 5×108
时限2s

这个题我们可以通过建立两棵平衡树来维护,对于MIN_SORT_GAP的答案,我们将每个数都加入平衡树,在加入时,我们找到加入的val的前驱和后继,而减数和被减数有val的减式中最小的值就是这两种情况之一,我们取最小值就好啦,而对于MIN_GAP的答案,我们需要用链表来模拟添加的情况,然后在平衡树中减去原有的相邻的减值,然后在加上加入这个数之后相邻的数相减的情况得到的值。
代码:

#pragma GCC optimize(2)
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 7;
std::mt19937 rnd(233);
struct treap
{
    int size, val, l, r, key;
} fhq1[maxn<<5], fhq2[maxn<<5];
int a[maxn << 4], b[maxn << 4], c[maxn << 4];
int cnt1, cnt2, root1, root2;
int ans = 1e9 + 10;
inline int newnode(treap *t,int val,int &cnt)
{
    t[++cnt] = treap{1, val, 0, 0, rnd()};
    return cnt;
}
inline void update(treap *t,int cnt)
{
    t[cnt].size = t[t[cnt].l].size + t[t[cnt].r].size + 1;
}
inline void split(treap *t,int now,int val,int &x,int &y)
{
    if(!now){
        x = y = 0;
        return;
    }
    if(t[now].val<=val){
        x = now;
        split(t, t[now].r, val, t[now].r, y);
    }
    else{
        y = now;
        split(t, t[now].l, val, x, t[now].l);
    }
    update(t, now);
}
inline int merge(treap*t,int x,int y)
{
    if(!x||!y){
        return x + y;
    }
    if(t[x].key>=t[y].key){
        t[x].r = merge(t, t[x].r, y);
        update(t, x);
        return x;
    }
    else{
        t[y].l = merge(t, x, t[y].l);
        update(t, y);
        return y;
    }
}
int x, y, z;
inline int pre(treap *t,int root)
{
    while(t[root].l)
        root = t[root].l;
    return t[root].val;
}
inline int nex(treap *t,int root)
{
    while(t[root].r)
        root = t[root].r;
    return t[root].val;
}
inline void insert(treap *t,int &root,int val,int &cnt)
{
    split(t, root, val, x, y);
    if(t==fhq1){
        if(t[x].size){
            ans = min(ans, abs(val - nex(t, x)));
        }
        if(t[y].size){
            ans = min(ans, abs(val - pre(t, y)));
        }
    }
    root = merge(t,merge(t,x, newnode(t, val, cnt)), y);
}
inline void del(treap *t,int &root,int val)
{
    split(t, root, val, x, z);
    split(t, x, val - 1, x, y);
    y = merge(t, t[y].l, t[y].r);
    root = merge(t, merge(t, x, y), z);
}
inline void read(int &val)
{
    int f = 1;
    val = 0;
    char c = getchar();
    while(c<'0' || c>'9'){
        if(c=='-')
            f = -1;
        c = getchar();
    }
    while(c>='0' && c<='9'){
        val = val * 10 + c - '0';
        c = getchar();
    }
    val *= f;
}
int len;
char arr[100];
int main()
{
    int n, m;
    read(n), read(m);
    for (register int i = 1; i <= n;i++){
        read(a[i]);
        insert(fhq1, root1, a[i], cnt1);
    }
    for (register int i = 2; i <= n;i++){
        insert(fhq2, root2, abs(a[i] - a[i - 1]), cnt2);
        b[i] = i - 1;
    }
    b[n + 1] = n;
    for (register int i = 1; i < n;i++){
        c[i] = i + 1;
    }
    len = n;
    while(m--)
    {
        int num, val;
        scanf("%s", arr);
        if(arr[0]=='I'){
            read(num), read(val);
            if(num<n){
                del(fhq2, root2, abs(a[num + 1] - a[b[num + 1]]));
            }
            insert(fhq2, root2, abs(val - a[b[num + 1]]), cnt2);
            if(num<n)
                insert(fhq2, root2, abs(val-a[num+1]), cnt2);
            a[++len] = val;
            b[len] = b[num + 1];
            c[len] = num + 1;
            c[b[num + 1]] = len, b[num + 1] = len;
            insert(fhq1, root1, val, cnt1);
        }
        else{
            if(arr[4]=='G')
                printf("%d\n", pre(fhq2, root2));
            else
                printf("%d\n", ans);
        }
    }
}

最后提一句,这道题会卡常,最后两个点一直过不去,一直T着,最后没办法,气急败坏之下开了氧气飘过[雾]。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值