BZOJ1058 || 洛谷P1110 [ZJOI2007]报表统计【平衡树】

Time Limit: 15 Sec
Memory Limit: 162 MB

Description

小Q的妈妈是一个出纳，经常需要做一些统计报表的工作。今天是妈妈的生日，小Q希望可以帮妈妈分担一些工作，作为她的生日礼物之一。经过仔细观察，小Q发现统计一张报表实际上是维护一个可能为负数的整数数列，并且进行一些查询操作。在最开始的时候，有一个长度为N的整数序列，并且有以下三种操作：
　　 INSERT i k 在原数列的第i个元素后面添加一个新元素k； 如果原数列的第i个元素已经添加了若干元素，则添加在这些元素的最后（见下面的例子）
　　 MIN_GAP 查询相邻两个元素的之间差值（绝对值）的最小值
　　 MIN_SORT_GAP 查询所有元素中最接近的两个元素的差值（绝对值）
例如一开始的序列为 5 3 1 执行操作INSERT 2 9将得到： 5 3 9 1 此时MIN_GAP为2，MIN_SORT_GAP为2。 再执行操作INSERT 2 6将得到： 5 3 9 6 1 注意这个时候原序列的第2个元素后面已经添加了一个9，此时添加的6应加在9的后面。这个时候MIN_GAP为2，MIN_SORT_GAP为1。于是小Q写了一个程序，使得程序可以自动完成这些操作，但是他发现对于一些大的报表他的程序运行得很慢，你能帮助他改进程序么？

Input

第一行包含两个整数N，M，分别表示原数列的长度以及操作的次数。第二行为N个整数，为初始序列。接下来的M行每行一个操作，即“INSERT i k”，“MIN_GAP”，“MIN_SORT_GAP”中的一种（无多余空格或者空行）。

Output

对于每一个“MIN_GAP”和“MIN_SORT_GAP”命令，输出一行答案即可。

HINT

N , M ≤500000 对于所有的数据，序列内的整数不超过5*10^8。

---

题目分析

一开始看错题意以为I操作要找新数列的位置还写了个splay
由于插入操作是找原数列的位置
那么也就相当于维护n段数字，而每段只关心开头和末尾的两个数字

开两个平衡树
第一个用来保存当前数列中所有数
每次插入新数 k 时，在这颗平衡树里查询k的前驱后继，更新询问3的答案

第二个平衡树维护相邻的两个数的差值
一开始先插入原数列相邻两个数的差值
对于$Iik$操作，删除第$i$段末尾的数字与第$i+1$段开头的数字的差值
再分别插入k与这两个数的差值，询问时直接输出平衡树的最小值即可

最后来个小优化，如果所有数字差值的最小值等于0，那么这个答案就不会再变了，所以也不用再更新第一颗平衡树了
虽然随手造个特殊数据就可以卡掉这个无脑优化，但对随机数据还挺有效的

年迈的niiick越发懒惰，写Treap什么的是不可能了，能用set就用set

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<cmath>
#include<set>
using namespace std;
typedef long long lt;
typedef double dd;
#define IT set<int>::iterator

int read()
{
    int x=0,f=1;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return x*f;
}

const int inf=1e9;
const int maxn=1000010;
int n,m,ans=1e9;
int st[maxn],ed[maxn];
char ss[20];
multiset<int> mi,arry;

void update(int x)
{
    IT it=arry.lower_bound(x);
    if(it!=arry.end()) ans=min(ans,*it-x);
    if(it!=arry.begin()) --it,ans=min(ans,x-(*it)); 
    arry.insert(x);
}

void ins(int pos,int x)
{
    if(ans!=0) update(x);
    mi.insert(abs(x-ed[pos]));
    if(pos!=n)
    {
        mi.erase(mi.find(abs(st[pos+1]-ed[pos])));
        mi.insert(abs(st[pos+1]-x));
    }
    ed[pos]=x;
}

int main()
{
    n=read();m=read();
    
    for(int i=1;i<=n;++i) 
    st[i]=ed[i]=read();
    
    for(int i=1;i<n;++i)
    mi.insert(abs(st[i+1]-ed[i]));
    
    arry.insert(-inf); arry.insert(inf);
    for(int i=1;i<=n;++i) 
    if(ans!=0) update(st[i]);
    
    while(m--)
    {
    	scanf("%s",&ss);
    	if(ss[0]=='I'){ int k=read(),val=read(); ins(k,val);}
        else if(ss[0]=='M'&&ss[4]=='S') printf("%d\n",ans);
        else if(ss[0]=='M'&&ss[4]=='G') printf("%d\n",*mi.begin());
    }/**/
    return 0;
}