您的位置: 首页  >  IT文章  >  [bzoj3123][sdoi2013森林] (树上主席树+lca+并查集启发式合并+暴力重构森林)

[bzoj3123][sdoi2013森林] (树上主席树+lca+并查集启发式合并+暴力重构森林)

分类: IT文章 • 2022-05-16 11:08:02

Description

[bzoj3123][sdoi2013森林] (树上主席树+lca+并查集启发式合并+暴力重构森林)

Input

第一行包含一个正整数testcase，表示当前测试数据的测试点编号。保证1≤testcase≤20。 
第二行包含三个整数N，M，T，分别表示节点数、初始边数、操作数。第三行包含N个非负整数表示 N个节点上的权值。 
 接下来 M行，每行包含两个整数x和 y，表示初始的时候，点x和点y 之间有一条无向边, 接下来 T行，每行描述一个操作，格式为“Q x y k”或者“L x y ”，其含义见题目描述部分。

Output

对于每一个第一类操作，输出一个非负整数表示答案。 

Sample Input

1 
8 4 8
1 1 2 2 3 3 4 4
4 7
1 8
2 4
2 1
Q 8 7 3 Q 3 5 1 
Q 10 0 0 
L 5 4 
L 3 2 L 0 7 
Q 9 2 5 Q 6 1 6

Sample Output

2 
2
1
4
2

HINT

对于第一个操作 Q 8 7 3，此时 lastans=0，所以真实操作为Q 8^0 7^0 3^0，也即Q 8 7 3。点8到点7的路径上一共有5个点，其权值为4 1 1 2 4。这些权值中，第三小的为 2，输出 2，lastans变为2。对于第二个操作 Q 3 5 1 ，此时lastans=2，所以真实操作为Q 3^2 5^2 1^2 ，也即Q 1 7 3。点1到点7的路径上一共有4个点，其权值为 1 1 2 4 。这些权值中，第三小的为2，输出2，lastans变为 2。之后的操作类似。 

[bzoj3123][sdoi2013森林] (树上主席树+lca+并查集启发式合并+暴力重构森林)

Solution

时限比较长暗示此题的解法比较暴力

和前面count on a tree的做法一样，先遍历整片森林，初始化倍增数组，把点投到可持久化线段树里去

对于询问操作，一样地，直接递归求解即可

对于连接操作，我们用并查集加size域启发式合并来处理森林的联通状况，方便我们重构树的时候减少重构的点的数量，这样就优化了暴力重构的时间

#include<stdio.h>
#include<string.h>
#define N 80110
#define INF 1000000000
#define mid ((x>>1)+(y>>1)+(x&y&1))
inline void exc(int &x,int &y){
    x^=y;y^=x;x^=y;
}
inline int Rin(){
    int x=0,c=getchar(),f=1;
    for(;c<48||c>57;c=getchar())
        if(!(c^45))f=-1;
    for(;c>47&&c<58;c=getchar())
        x=(x<<1)+(x<<3)+c-48;
    return x*f;
}
int n,m,T,val[N],jump[N][20],dep[N],pb[N],top,ans;
struct st{int f,s;}s[N];
inline int pre(int x){
    while(s[x].f^x)x=s[x].f;
    return x;
}
inline void onion(int x,int y){
    x=pre(x),y=pre(y);
    s[y].f=x,s[x].s+=s[y].s;
}
struct pt{int v;pt *nxt;}
*fst[N],e[N<<1],*tot=e;
inline void link(int x,int y){
    *++tot=(pt){y,fst[x]},fst[x]=tot;
    *++tot=(pt){x,fst[y]},fst[y]=tot;
}
struct nt{
    nt *l,*r;
    int s;
}*rt[N],pool[20002000],*C=pool;
inline nt *newnt(nt *_,nt *__,int ___){
    C->l=_;C->r=__;C->s=___;
    return C++;
}
nt *build(nt *p,int x,int y,int k){
    if(!(x^y))return newnt(0x0,0x0,p->s+1);
    if(k<=mid)return newnt(build(p->l,x,mid,k),p->r,p->s+1);
    return newnt(p->l,build(p->r,mid+1,y,k),p->s+1);
}
void dfs(int x){
    dep[x]=dep[jump[x][0]]+1;
    rt[x]=build(rt[jump[x][0]],0,INF,val[x]);
    for(pt *j=fst[x];j;j=j->nxt)
        if(j->v^jump[x][0])
            jump[j->v][0]=x,
            dfs(j->v);
}
void dfs(int x,int f){
    pb[++top]=x;
    jump[x][0]=f;
    dep[x]=dep[f]+1;
    rt[x]=build(rt[f],0,INF,val[x]);
    for(pt *j=fst[x];j;j=j->nxt)
        if(j->v^f)dfs(j->v,x);
}
int lca(int x,int y){
    if(dep[x]<dep[y])exc(x,y);
    for(int j=19;~j;j--)
        if(dep[jump[x][j]]>=dep[y])
            x=jump[x][j];
    if(!(x^y))return x;
    for(int j=19;~j;j--)
        if(jump[x][j]^jump[y][j])
            x=jump[x][j],y=jump[y][j];
    return jump[x][0];
}
int secret(nt *p1,nt *p2,nt *p3,nt *p4,int x,int y,int k){
    if(!(x^y))return x;
    int c=p1->l->s+p2->l->s-p3->l->s-p4->l->s;
    if(k<=c)return secret(p1->l,p2->l,p3->l,p4->l,x,mid,k);
    return secret(p1->r,p2->r,p3->r,p4->r,mid+1,y,k-c);
}
int feel(int x,int y,int k){
    int t=lca(x,y);
    return secret(rt[x],rt[y],rt[t],rt[jump[t][0]],0,INF,k);
}
int main(){
    T=Rin(),n=Rin(),m=Rin(),T=Rin();
    for(int i=1;i<=n;i++)
        s[i].f=i,s[i].s=1;
    for(int i=1;i<=n;i++)
        val[i]=Rin();
    for(int x,y;m;m--)
        x=Rin(),y=Rin(),link(x,y),onion(x,y);
    rt[0]=newnt(C,C,0);
    for(int i=1;i<=n;i++)
        if(!jump[i][0])
            dfs(i);
    for(int j=1;j<=19;j++)
        for(int i=1;i<=n;i++)
            jump[i][j]=jump[jump[i][j-1]][j-1];
    char sign[6];
    for(int x,y,k;T;T--){
        scanf("%s",sign);
        x=Rin()^ans,y=Rin()^ans;
        if(sign[0]=='Q'){
            k=Rin()^ans;
            printf("%d
",ans=feel(x,y,k));
        }
        else{
            if(s[pre(x)].s>s[pre(y)].s)
                exc(x,y);
            top=0;
            dfs(x,y);
            onion(x,y);
            for(int j=1;j<=19;j++)
                for(int i=1;i<=top;i++)
                    jump[pb[i]][j]=jump[jump[pb[i]][j-1]][j-1];
            link(x,y);
        }
    }
    return 0;
}

相关推荐