bzoj1058: [ZJOI2007]报表统计（splay+堆）

题目传送门

。

解法：

第一问就只需用小根堆维护每个原序列最后一个插入的是什么就行了。

每次进来的先把原来这个位置最后一个跟后面那个位置的差先删了

然后把新的两个差值加入堆。

第二问就是splay维护前驱后继就行了呀。

代码实现：

#include<queue>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
struct heap{
priority_queue<int,vector<int>,greater<int> >A,B;
void push(int x){A.push(x);}
void erase(int x){B.push(x);}
void pop(){while(B.size()&&A.top()==B.top())A.pop(),B.pop();A.pop(); }
int top(){
while(B.size()&&A.top()==B.top())A.pop(),B.pop();
if(!A.size())return 0;return A.top();
}
}A;
priority_queue<int,vector<int>,greater<int> >B;
struct node {int n,c,son[2],d,f;}tr[1100000];int len,root;
void add(int d,int f) {
len++;
tr[len].f=f;tr[len].c=tr[len].n=1;
tr[len].d=d;tr[len].son[0]=tr[len].son[1]=0;
if(d<tr[f].d)tr[f].son[0]=len;
else tr[f].son[1]=len;
}
int update(int now) {int lc=tr[now].son[0],rc=tr[now].son[1];tr[now].c=(tr[lc].c+tr[rc].c+tr[now].n);}
void rotate(int x,int w) {
int f=tr[x].f,ff=tr[f].f;int R,r;
R=f;r=tr[x].son[w];tr[R].son[1-w]=r;if(r!=0)tr[r].f=R;
R=ff;r=x;if(tr[R].son[0]==f)tr[R].son[0]=r;else tr[R].son[1]=r;tr[r].f=R;
R=x;r=f;tr[R].son[w]=r;tr[r].f=R;
update(f);update(x);
}
void splay(int x,int rt) {
while(tr[x].f!=rt) {
int f=tr[x].f,ff=tr[f].f;
if(ff==rt) {if(tr[f].son[0]==x)rotate(x,1);else rotate(x,0);}
else {
if(tr[ff].son[0]==f && tr[f].son[0]==x){rotate(f,1);rotate(x,1);}
else if(tr[ff].son[1]==f && tr[f].son[0]==x){rotate(x,1);rotate(x,0);}
else if(tr[ff].son[1]==f && tr[f].son[1]==x){rotate(f,0);rotate(x,0);}
else rotate(x,0),rotate(x,1);
}
}
if(rt==0)root=x;
}
int findip(int d) {
int x=root;
while(tr[x].d!=d) {
if(d<tr[x].d) {if(tr[x].son[0]==0)break;x=tr[x].son[0];}
else {if(tr[x].son[1]==0)break;x=tr[x].son[1];}
}return x;
}
void ins(int d) {
int x=findip(d);
if(len==0){add(d,0);root=len;return ;}
if(tr[x].d==d){tr[x].n++;update(x);return ;}
else{add(d,x);update(len);update(x);}
}
int findqianqu(int d) {
int x=findip(d);
if(d==tr[x].d)return tr[x].d;splay(x,0);
if(tr[x].son[0]==0)return 1073741824;
if(d<=tr[x].d){x=tr[x].son[0];while(tr[x].son[1]!=0) x=tr[x].son[1];}
return tr[x].d;
}
int findhouji(int d)
{
int x=findip(d);
if(d==tr[x].d)return tr[x].d;splay(x,0);
if(tr[x].son[1]==0)return 1073741824;
if(tr[x].d<=d) {x=tr[x].son[1];while(tr[x].son[0]!=0) x=tr[x].son[0];}
return tr[x].d;
}
int a[510000],last[510000];
char s[20];int b[510000];
int main() {
//freopen("1058.in","r",stdin);freopen("1058.out","w",stdout);
int n,m;scanf("%d%d",&n,&m);len=0;root=0;
for(int i=1;i<=n;i++) {scanf("%d",&a[i]);ins(a[i]);}
for(int i=1;i<=n;i++)last[i]=a[i];
for(int i=1;i<n;i++)A.push(abs(a[i]-a[i+1]));
for(int i=1;i<=n;i++)b[i]=a[i];
sort(b+1,b+1+n);ll ans=1<<30;
for(int i=2;i<=n;i++)ans=min(ans,ll(b[i]-b[i-1]));
for(int i=1;i<=m;i++) {
scanf("%s",s+1);
if(s[1]=='I') {
int x,k;scanf("%d%d",&x,&k);
if(x!=n)A.erase(abs(last[x]-a[x+1]));
A.push(abs(k-last[x]));if(x!=n)A.push(abs(k-a[x+1]));last[x]=k;
int q,h;q=findqianqu(k);h=findhouji(k);
if(q!=1073741824)ans=min(ans,ll(k-q));
if(h!=1073741824)ans=min(ans,ll(h-k));ins(k);
}
else if(s[5]=='G')printf("%d\n",A.top());
else printf("%lld\n",ans);
}
return 0;
}