bzoj1058: [ZJOI2007]報表統計(splay+堆)

Hanks_o發表於2018-03-03

題目傳送門

解法:
第一問就只需用小根堆維護每個原序列最後一個插入的是什麼就行了。
每次進來的先把原來這個位置最後一個跟後面那個位置的差先刪了
然後把新的兩個差值加入堆。

第二問就是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;
}

相關文章