bzoj 3123 可持久化线段树启发式合并

  首先没有连边的操作的时候,我们可以用可持久化线段树来维护这棵树的信息,建立权值可持久化线段树,那么每个点继承父节点的线段树,当询问为x,y的时候我们可以询问rot[x]+rot[y]-rot[lca(x,y)]-rot[lca(x,y)->father]这棵树来得知这个链的信息。

  那么对于连边操作,相当于合并两棵树,我们可以将树的节点数小的树全部拆掉连到节点大的树中,这样每个点最多会被操作logn次,每次操作的时间复杂度为logn,所以是mlog^2n的。

  反思:对于树的连通性我是用并查集维护的,对于合并操作还需要dfs一次小的树来维护各种信息,但是忘记对x,y点连边了,导致一直RE.(RE是因为某次值不正确,导致下一次^的点超过n)。

     各种慢= =。

/**************************************************************
    Problem: 3123
    User: BLADEVIL
    Language: C++
    Result: Accepted
    Time:16612 ms
    Memory:81512 kb
****************************************************************/
 
//By BLADEVIL
#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 100010
#define maxm 200010
 
using namespace std;
 
struct rec {
    int key,num;
    rec() {
        key=num=0;
    }
}a[maxn];
 
struct segment {
    int left,right,cnt;
    int son[2];
    segment() {
        left=right=cnt=0;
        memset(son,0,sizeof son);
    }
}t[maxn<<5];
 
int n,m,l,N,tot;
int pre[maxm<<1],last[maxm<<1],other[maxm<<1];
int ans[maxn],rot[maxn];
int dep[maxn],jump[maxn][20],father[maxn],size[maxn],que[maxn];
 
void connect(int x,int y) {
    pre[++l]=last[x];
    last[x]=l;
    other[l]=y;
}
 
int getfather(int x) {
    if (father[x]==x) return x;
    return father[x]=getfather(father[x]);
}
 
void dfs(int x,int fa) {
    jump[x][0]=fa; dep[x]=dep[fa]+1;
    for (int p=last[x];p;p=pre[p]) {
        if (other[p]==fa) continue;
        dfs(other[p],x);
    }
}
 
bool cmp1(rec x,rec y) {
    return x.key<y.key;
}
 
bool cmp2(rec x,rec y) {
    return x.num<y.num;
}
 
int lca(int x,int y) {
    if (dep[x]>dep[y]) swap(x,y);
    int d=dep[y]-dep[x];
    for (int i=0;i<=16;i++) if (d&(1<<i)) y=jump[y][i];
    if (x==y) return x;
    for (int i=16;i>=0;i--) if (jump[x][i]!=jump[y][i]) x=jump[x][i],y=jump[y][i];
    return jump[x][0];
}
 
void build(int &x,int l,int r) {
    if (!x) x=++tot;
    t[x].left=l; t[x].right=r;
    if (t[x].left==t[x].right) return ;
    int mid=t[x].left+t[x].right>>1;
    build(t[x].son[0],l,mid); build(t[x].son[1],mid+1,r);
}
 
void insert(int &x,int rot,int key) {
    if (!x) x=++tot;
    t[x].left=t[rot].left; t[x].right=t[rot].right;
    if (t[x].left==t[x].right) {
        t[x].cnt=t[rot].cnt+1;
        return ;
    }
    int mid=t[x].left+t[x].right>>1;
    if (key>mid) {
        t[x].son[0]=t[rot].son[0];
        insert(t[x].son[1],t[rot].son[1],key);
    } else {
        t[x].son[1]=t[rot].son[1];
        insert(t[x].son[0],t[rot].son[0],key);
    }
    t[x].cnt=t[rot].cnt+1;
}
 
int query(int x,int y,int l1,int l2,int k) {
    //printf("%d %d %d ",x,t[x].left,t[x].right);
    if (t[x].left==t[x].right) return t[x].left;
    int cur=t[t[x].son[0]].cnt+t[t[y].son[0]].cnt-t[t[l1].son[0]].cnt-t[t[l2].son[0]].cnt;
    //printf("%d\n",cur);
    if (k>cur) return query(t[x].son[1],t[y].son[1],t[l1].son[1],t[l2].son[1],k-cur); else
        return query(t[x].son[0],t[y].son[0],t[l1].son[0],t[l2].son[0],k);
}
 
void make(int x,int fa) {
    insert(rot[x],rot[fa],a[x].key);
    for (int p=last[x];p;p=pre[p]) {
        if (other[p]==fa) continue;
        make(other[p],x);
    }
}
 
void update(int x,int fa,int j) {
    jump[x][j]=jump[jump[x][j-1]][j-1];
    for (int p=last[x];p;p=pre[p]) {
        if (other[p]==fa) continue;
        update(other[p],x,j);
    }
}
 
void merge(int x,int y) {
    int fx=getfather(x),fy=getfather(y);
    //if (x==1) printf("%d %d\n",fx,fy);
    if (size[fx]<size[fy]) swap(x,y);
    fx=getfather(x),fy=getfather(y);
    father[fy]=fx; size[fx]+=size[fy];
    dfs(y,x); make(y,x); 
    for (int i=1;i<=16;i++) update(y,x,i);
    //for (int i=1;i<=n;i++) printf("%d %d\n",i,jump[i][0]);
}
 
int main() {
    //freopen("3123.in","r",stdin); freopen("3123.out","w",stdout);
    int task; scanf("%d",&task);
    scanf("%d%d%d",&n,&m,&task);
    for (int i=1;i<=n;i++) scanf("%d",&a[a[i].num=i].key); 
 
    sort(a+1,a+1+n,cmp1);
    int cur; ans[1]=cur=a[1].key;
    for (int i=1,j=1;i<=n;i++) 
        if (a[i].key==cur) a[i].key=j; else ans[++j]=cur=a[i].key,a[i].key=j,N=j;
    sort(a+1,a+1+n,cmp2);
    //for (int i=1;i<=n;i++) printf("%d ",ans[i]); printf("\n");
    //for (int i=1;i<=n;i++) printf("%d ",a[i].key); printf("\n");
     
    for (int i=1;i<=n;i++) father[i]=i,size[i]=1;
    for (int i=1;i<=m;i++) {
        int x,y; scanf("%d%d",&x,&y);
        connect(x,y); connect(y,x);
        int fx=getfather(x),fy=getfather(y);
        father[fx]=fy; size[fy]+=size[fx];
    }
    for (int i=1;i<=n;i++) if (!jump[i][0]) dfs(i,0);
    //for (int i=1;i<=n;i++) printf("%d %d\n",i,father[i]);
    //for (int i=1;i<=n;i++) if (i==getfather(i)) printf("%d %d\n",i,size[i]);
     
    for (int j=1;j<=16;j++)
        for (int i=1;i<=n;i++) jump[i][j]=jump[jump[i][j-1]][j-1];
         
    build(rot[0],1,N);
    for (int i=1;i<=n;i++) if (!jump[i][0]) make(i,0);
     
    int ANS=0; char Q[3];
    while (task--) {
        scanf("%s",Q);
        if (Q[0]=='Q') {
            int x,y,k; scanf("%d%d%d",&x,&y,&k);
            x^=ANS; y^=ANS; k^=ANS;
            //printf("|%d %d %d\n",x,y,k);
            int rr=lca(x,y); //printf("%d\n",rr);
            //for (int i=1;i<=n;i++) printf("%d %d\n",i,jump[i][0]);
            printf("%d\n",ANS=ans[query(rot[x],rot[y],rot[rr],rot[jump[rr][0]],k)]);
            //for (int i=1;i<=n;i++) if (i==getfather(i)) printf("%d %d\n",i,size[i]);
        } else {    
            int x,y; scanf("%d%d",&x,&y);
            x^=ANS; y^=ANS; connect(x,y); connect(y,x);
            //printf("|%d %d\n",x,y);
            merge(x,y);
        }
    }
    return 0;
}

 

 

posted on 2014-04-21 12:56  BLADEVIL  阅读(851)  评论(0编辑  收藏  举报