bzoj 1058: [ZJOI2007]报表统计（splay+线段树）

1058: [ZJOI2007]报表统计

Time Limit: 15 Sec   Memory Limit: 162 MB
Submit: 2816   Solved: 969
[ Submit][ Status][ Discuss]

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”命令，输出一行答案即可。

Sample Input

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

Sample Output

2
2
1

HINT

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

Source

[ Submit][ Status][ Discuss]

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#define N 1000003
#define inf 0x3fffffff
using namespace std;
int n,m,tot,ans,root,cnt;
struct data
{
    int l,r,num;
};data line[N];
int tree[N*4],a[N*2],ch[N*2][3],fa[N*2],key[N*2],size[N*2];
void update(int now)
{
    tree[now]=min(tree[now<<1],tree[now<<1|1]);
}
void build(int now,int l,int r)
{
    if(l==r)  {
       tree[now]=a[l];
       return;
    }
    int mid=(l+r)/2;
    build(now<<1,l,mid);
    build(now<<1|1,mid+1,r);
    update(now);
}
void pointchange(int now,int l,int r,int x,int v)
{
    if(l==r){
        tree[now]=v;
        return;
    }
    int mid=(l+r)/2;
    if (x<=mid)  pointchange(now<<1,l,mid,x,v);
    else pointchange(now<<1|1,mid+1,r,x,v);
    update(now);
}
int get(int x)
{
    return ch[fa[x]][1]==x;
}
void rotate(int x)
{
   int y=fa[x]; int z=fa[y]; int which=get(x);
   ch[y][which]=ch[x][which^1]; fa[ch[x][which^1]]=y;
   ch[x][which^1]=y; fa[y]=x; fa[x]=z;
   if (z)  ch[z][ch[z][1]==y]=x;    
}
void splay(int x)
{
    for (int f;(f=fa[x]);rotate(x))
     if (fa[f])
      rotate(get(f)==get(x)?f:x);
    root=x;
}
void insert(int x)
{
    if (!root)
    {
        root=++cnt; key[cnt]=x; fa[cnt]=ch[cnt][1]=ch[cnt][0]=0; size[cnt]++;
        return;
    }
    int now=root,f;
    while (true)
    {
        if (key[now]==x){
            size[now]++; splay(now); return;
        }
        f=now;
        now=ch[now][x>key[now]];
        if(!now)
        {
            cnt++; fa[cnt]=f; ch[f][x>key[f]]=cnt; size[cnt]++;
            key[cnt]=x; splay(cnt); return;
        }
    }
}
int pre()
{
    int now=ch[root][0];
    if (size[root]>1) return root;
    while(ch[now][1]) now=ch[now][1];
    return now;
}
int next()
{
    int now=ch[root][1];
    if (size[root]>1)  return root;
    while(ch[now][0]) now=ch[now][0];
    return now;
}
int main()
{
    scanf("%d%d",&n,&m);  ans=inf;
    for (int i=1;i<=n;i++)
     {
        int x; scanf("%d",&x);
        line[i].l=x; line[i].r=x; line[i].num=inf;
        a[i]=line[i].num;
        insert(x);
        int l=pre();  int r=next();
        //cout<<key[l]<<" "<<x<<" "<<key[r]<<endl;
        if (l)   ans=min(ans,abs(key[l]-x));
        if (r)   ans=min(ans,abs(key[r]-x));
     }
    for (int i=2;i<=n;i++)
     a[n+i-1]=abs(line[i].l-line[i-1].r);
    tot=2*n-1;
    build(1,1,tot);
    for (int i=1;i<=m;i++)
    {
        char s[20]; scanf("%s",s); 
        if (s[0]=='I')  {
            int x,y; scanf("%d%d",&x,&y); 
            if (abs(y-line[x].r)<line[x].num)
             {
                line[x].num=abs(y-line[x].r);
                pointchange(1,1,tot,x,line[x].num);
             }
             line[x].r=y;
             if (x!=n)  pointchange(1,1,tot,n+x,abs(line[x].r-line[x+1].l));
             insert(y);
             int l=pre();  int r=next();
             if (l)   ans=min(ans,abs(key[l]-y));
             if (r)   ans=min(ans,abs(key[r]-y));
        }
        else  if (strlen(s)==7)  printf("%d\n",tree[1]);
        else  printf("%d\n",ans);
    }
}