hdu3726 Graph and Queries(离线+平衡树启发式合并)

我们把操作离线,删边操作就变成了合并。每个连通块维护一棵splay,合并时把sz小的暴力插到sz大的里面去即可。复杂度 O(nlog2n) ,为什么呢,因为一个点x被暴力插入一次,则它的sz至少大了一倍,所以一个点最多暴力插入logn次。
tips:ins之后要转到跟呀转到跟!

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 20010
#define M 60010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,m,nq,rt[N],pa[N],fa[N],c[N][2],sz[N],v[N],qq[N],tst=0;
struct Edge{
    int x,y;
}e[M];
bool dele[M];
struct event{
    int op,x,k;
}q[500000];
inline int find(int x){return x==pa[x]?x:pa[x]=find(pa[x]);}
inline int find(int p,int x){
    if(x==sz[c[p][0]]+1) return v[p];
    if(x<=sz[c[p][0]]) return find(c[p][0],x);
    return find(c[p][1],x-sz[c[p][0]]-1);
}
inline void update(int x){
    int l=c[x][0],r=c[x][1];
    sz[x]=sz[l]+sz[r]+1;
}
inline void rotate(int x,int &k){
    int y=fa[x],z=fa[y],l=x==c[y][1],r=l^1;
    if(y==k) k=x;
    else c[z][y==c[z][1]]=x;
    fa[c[x][r]]=y;fa[y]=x;fa[x]=z;
    c[y][l]=c[x][r];c[x][r]=y;update(y);update(x);
}
inline void splay(int x,int &k){
    while(x!=k){
        int y=fa[x],z=fa[y];
        if(y!=k){
            if(x==c[y][1]^y==c[z][1]) rotate(x,k);
            else rotate(y,k);
        }rotate(x,k);
    }
}
inline void ins(int &p,int x,int Fa){
    if(!p){p=x;sz[p]=1;fa[p]=Fa;return;}
    if(v[x]<=v[p]) ins(c[p][0],x,p);
    else ins(c[p][1],x,p);
}
inline void merge(int x,int y){
    int xx=find(x),yy=find(y);
    if(xx==yy) return;if(sz[rt[xx]]>sz[rt[yy]]) swap(xx,yy);
    pa[xx]=yy;int h=1,t=0;qq[++t]=rt[xx];
    while(h<=t){
        int x=qq[h++];
        if(c[x][0]) qq[++t]=c[x][0];
        if(c[x][1]) qq[++t]=c[x][1];
    }for(int i=1;i<=t;++i){
        int x=qq[i];fa[x]=c[x][0]=c[x][1]=0;ins(rt[yy],x,0);splay(x,rt[yy]);
    }
}
inline void del(int x,int &rt){
    splay(x,rt);
    if(c[x][0]*c[x][1]==0){
        rt=c[x][0]+c[x][1];fa[rt]=0;c[x][0]=c[x][1]=0;return;
    }int pre=c[x][0],succ=c[x][1];
    while(c[pre][1]) pre=c[pre][1];
    while(c[succ][0]) succ=c[succ][0];
    splay(pre,rt);splay(succ,c[rt][1]);
    c[succ][0]=fa[x]=0;update(succ);update(pre);
}
int main(){
//  freopen("a.in","r",stdin);
    while(1){
        n=read();m=read();if(!n) break;nq=0;double ans=0;int ans1=0;++tst;
        for(int i=1;i<=n;++i) v[i]=read();
        for(int i=1;i<=m;++i) e[i].x=read(),e[i].y=read();
        while(1){
            char op[5];scanf("%s",op+1);if(op[1]=='E') break;
            int x=q[++nq].x=read();
            if(op[1]=='Q') q[nq].op=1,q[nq].k=read();
            else if(op[1]=='D') q[nq].op=2,dele[x]=1;
            else q[nq].op=3,q[nq].k=v[x],v[x]=read();
        }for(int i=1;i<=n;++i) rt[i]=i,pa[i]=i,sz[i]=1,fa[i]=c[i][0]=c[i][1]=0;
        for(int i=1;i<=m;++i) if(!dele[i]) merge(e[i].x,e[i].y);
        for(int i=nq;i>=1;--i){
            if(q[i].op==1){
                int x=find(q[i].x);ans1++;
                if(q[i].k>sz[rt[x]]||q[i].k<=0) continue;
                ans+=find(rt[x],sz[rt[x]]-q[i].k+1);
            }else if(q[i].op==3){
                int x=find(q[i].x);
                del(q[i].x,rt[x]);v[q[i].x]=q[i].k;
                ins(rt[x],q[i].x,0);splay(q[i].x,rt[x]);
            }else merge(e[q[i].x].x,e[q[i].x].y),dele[q[i].x]=0;
        }printf("Case %d: %.6lf\n",tst,ans/ans1);
    }return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值