P1110 [ZJOI2007]报表统计 平衡树+堆

题目描述

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

输入输出格式

输入格式:

第一行包含两个整数N,M,分别表示原数列的长度以及操作的次数。

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

接下来的M行每行一个操作,即“INSERT i k”,“MIN_GAP”,“MIN_SORT_GAP”中的一种(无多余空格或者空行)。

输出格式:

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

输入输出样例

输入样例#1:
3 5
5 3 1
INSERT 2 9
MIN_SORT_GAP
INSERT 2 6
MIN_GAP
MIN_SORT_GAP
输出样例#1:
2
2
1

说明

对于30%的数据,N ≤ 1000 , M ≤ 5000

对于100%的数据,N , M ≤500000

对于所有的数据,序列内的整数不超过5*108。

时限2s


 

看了题之后,我第一反映就是平衡树,但是我又像有没有什么简单的算法可能过吗?

因为他询问不具体,我就想用链表来做。

但是你会发现最后一种询问你不好回答,于是无奈之举,只能用平衡树来回答第三问。

因为平衡树可以找前驱后继。

还有,第二问可以用两个堆做。很好理解的。

最神奇的是,在luogu上测,我同一份代码,交了5次,两次内存超限,两次答案错误,最后一次对了。。。

代码:

 

#include<iostream>
#include<ctime>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
 
#define ll long long
#define il inline
#define db double
 
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define abs(x) ((x)>0?(x):-(x))

using namespace std;
 
il int gi()
{
    int x=0,y=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
        {
            if(ch=='-')
                y=-1;
            ch=getchar();
        }
    while(ch>='0'&&ch<='9')
        {
            x=x*10+ch-'0';
            ch=getchar();
        }
    return x*y;
}
 
int ans=2e9;
 
struct node
{
    int v;
    int l;
    node* ch[2];
}c[4000045],*rt,*pos=c;

il void newnode(node* &r,int v)
{
	r=pos++;
    r->v=v;
    r->l=rand();
    r->ch[0]=r->ch[1]=0;
}
 
il void roll(node* &r,bool t)
{
    node* y=r->ch[!t];
    r->ch[!t]=y->ch[t];
    y->ch[t]=r;
    r=y;
}
 
void ins(node* &r,int v)
{
    if(!r)
        newnode(r,v);
    else
        {
            bool t=r->v<v;
            ins(r->ch[t],v);
			if(r->ch[t]->l<r->l)
				roll(r,!t);
        }
}

int p1,p2;
 
void front(node* r,int v)
{
	if(!r)
		return;
	if(r->v>v)
		front(r->ch[0],v);
	else
		p1=r->v,front(r->ch[1],v);
}

void behind(node* r,int v)
{
	if(!r)
		return;
	if(r->v<v)
		behind(r->ch[1],v);
	else
		p2=r->v,behind(r->ch[0],v);
}

int heap[4000045],size;
 
il void put(int x)
{
    heap[++size]=x;
    int now=size,next;
    while(now!=1)
        {
            next=now>>1;
            if(heap[next]<=heap[now])
                break;
            swap(heap[now],heap[next]);
            now=next;
        }
}
 
il void pop()
{
    heap[1]=heap[size--];
    int now=1,next;
    while(now<<1<=size)
        {
            next=now<<1;
            if(heap[next+1]<heap[next]&&next<size)
                next++;
            if(heap[next]>=heap[now])
                break;
            swap(heap[now],heap[next]);
            now=next;
        }
}
 
int delheap[4000045],delsize;
 
il void delput(int x)
{
    delheap[++delsize]=x;
    int now=delsize,next;
    while(now!=1)
        {
            next=now>>1;
            if(delheap[next]<=delheap[now])
                break;
            swap(delheap[now],delheap[next]);
            now=next;
        }
}
 
il void delpop()
{
    delheap[1]=delheap[delsize--];
    int now=1,next;
    while(now<<1<=delsize)
        {
            next=now<<1;
            if(delheap[next+1]<delheap[next]&&next<delsize)
                next++;
            if(delheap[next]>=delheap[now])
                break;
            swap(delheap[now],delheap[next]);
            now=next;
        }
}

int num[4000045];
 
int next[4000045];
 
char S[45];
 
int main()
{
	freopen("2.in","r",stdin);
	freopen("2.out","w",stdout);
	
    srand((int)time(0));
 
    int n=gi(),m=gi();
 
    for(int i=1;i<=n;i++)
        {
            num[i]=gi();
            if(i>1)
                put(abs(num[i]-num[i-1]));
			p1=-1,p2=-1;
			front(rt,num[i]);
			behind(rt,num[i]);
			if(p2!=-1)
				ans=min(ans,p2-num[i]);
			if(p1!=-1)
				ans=min(ans,num[i]-p1);
            ins(rt,num[i]);
        }
 
    int x,y;
    for(int i=1;i<=m;i++)
        {
            scanf("%s",S);
            if(S[4]=='R')
                {
                    x=gi(),y=gi();
					p1=-1,p2=-1;
					front(rt,y);
					behind(rt,y);
					if(p2!=-1)
						ans=min(ans,p2-y);
					if(p1!=-1)
						ans=min(ans,y-p1);
                    ins(rt,y);
                    if(next[x]==0)
                        put(abs(y-num[x]));
                    else
                        {
                            put(abs(y-next[x]));
                            delput(abs(num[x+1]-next[x]));
                        }
                    put(abs(num[x+1]-y));
                    next[x]=y;
                }
            if(S[4]=='G')
                {
                    while(heap[1]==delheap[1]&&size>0&&delsize>0)
                        {
                            pop();
                            delpop();
                        }
                    printf("%d\n",heap[1]);
                }
            if(S[4]=='S')
				{
					printf("%d\n",ans);
				}
        }
 
    return 0;
}

 

posted @ 2017-10-17 21:49  GSHDYJZ  阅读(169)  评论(0编辑  收藏  举报