AC日记——[Sdoi2013]森林 bzoj 3123

3123: [Sdoi2013]森林

Time Limit: 20 Sec  Memory Limit: 512 MB
Submit: 3216  Solved: 944
[Submit][Status][Discuss]

Description

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。之后的操作类似。

 

 

思路：

　　启发式合并+主席数+lca+并查集；

　　具体请脑补，复杂地（nlong^2n）；

 

 

来，上代码：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

#define maxn 80005

using namespace std;

struct TreeNodeType {
    int l,r,dis;
};
struct TreeNodeType tree[];

struct EdgeType {
    int v,next;
};
struct EdgeType edge[maxn<<];

int if_z,t,n,m,size,dis[maxn],head[maxn];
int cnt,f[maxn][],ft[maxn],deep[maxn];
int num[maxn],dis_[maxn],last,tot,root[maxn];

char Cget;

inline void in(int &now)
{
    now=,if_z=,Cget=getchar();
    while(Cget>''||Cget<'')
    {
        if(Cget=='-') if_z=-;
        Cget=getchar();
    }
    while(Cget>=''&&Cget<='')
    {
        now=now*+Cget-'';
        Cget=getchar();
    }
    now*=if_z;
}

inline void edge_add(int u,int v)
{
    cnt++;
    edge[cnt].v=v;
    edge[cnt].next=head[u];
    head[u]=cnt;
}

int find(int x)
{
    if(x==ft[x]) return x;
    ft[x]=find(ft[x]);
    return ft[x];
}

void tree_add(int pre,int &now,int l,int r,int to)
{
    now=++tot;
    tree[now].dis=tree[pre].dis+;tree[now].r=tree[pre].r;tree[now].l=tree[pre].l;
    if(l==r) return ;
    int mid=(l+r)>>;
    if(to<=mid)
    {
        tree_add(tree[pre].l,tree[now].l,l,mid,to);

    }
    else
    {
        tree_add(tree[pre].r,tree[now].r,mid+,r,to);

    }
}

void search(int now,int pre)
{
    f[now][]=pre,deep[now]=deep[pre]+;
    for(int i=;i<=;i++) f[now][i]=f[f[now][i-]][i-];
    int pos=lower_bound(dis_+,dis_+size+,dis[now])-dis_;
    tree_add(root[pre],root[now],,size,pos);
    for(int i=head[now];i;i=edge[i].next)
    {
        if(edge[i].v==pre) continue;
        search(edge[i].v,now);
    }
}

void tree_build(int &now,int l,int r)
{
    now=++tot;
    if(l==r) return ;
    int mid=(l+r)>>;
    tree_build(tree[now].l,l,mid);
    tree_build(tree[now].r,mid+,r);
}

int get_same(int u,int x)
{
    for(int i=;i<=;i++) if(x&(<<i)) u=f[u][i];
    return u;
}

int lca(int x,int y)
{
    if(deep[x]<deep[y]) swap(x,y);
    x=get_same(x,deep[x]-deep[y]);
    if(x==y) return x;
    for(int i=;i>=;i--)
    {
        if(f[x][i]!=f[y][i])
        {
            x=f[x][i];
            y=f[y][i];
        }
    }
    return f[x][];
}

int tree_query(int pre,int fpre,int now,int now_,int k,int l,int r)
{
    if(l==r) return l;
    int pos=tree[tree[now].l].dis+tree[tree[now_].l].dis;
    pos-=(tree[tree[pre].l].dis+tree[tree[fpre].l].dis);
    int mid=(l+r)>>;
    if(k>pos) return tree_query(tree[pre].r,tree[fpre].r,tree[now].r,tree[now_].r,k-pos,mid+,r);
    else return tree_query(tree[pre].l,tree[fpre].l,tree[now].l,tree[now_].l,k,l,mid);
}

int main()
{
//  freopen("forest.in","r",stdin);
//  freopen("forest.out","w",stdout);
    in(t),in(n),in(m),in(t);
    for(int i=;i<=n;i++)
    {
        in(dis[i]);
        num[i]=;
        dis_[i]=dis[i],ft[i]=i;
    }
    sort(dis_+,dis_+n+);
    size=unique(dis_+,dis_+n+)-dis_-;
    int u,v,x,y,k;
    while(m--)
    {
        in(u),in(v);
        edge_add(u,v);
        edge_add(v,u);
        x=find(u),y=find(v);
        if(num[x]>num[y])
        {
            ft[y]=x;
            num[x]+=num[y];
        }
        else
        {
            ft[x]=y;
            num[y]+=num[x];
        }
    }
    tree_build(root[],,size);
    for(int i=;i<=n;i++)
    {
        if(ft[i]==i) search(i,);
    }
    char ch[];
    while(t--)
    {
        scanf("%s",ch);
        if(ch[]=='L')
        {
            in(u),in(v);
            u=u^last,v=v^last;
            edge_add(u,v);
            edge_add(v,u);
            x=find(u),y=find(v);
            if(num[x]>num[y])
            {
                ft[y]=x;
                num[x]+=num[y];
                search(v,u);
            }
            else
            {
                ft[x]=y;
                num[y]+=num[x];
                search(u,v);
            }
        }
        else
        {
            in(u),in(v),in(k);
            u=u^last,v=v^last,k=k^last;
            int lca_=lca(u,v);
            last=tree_query(root[lca_],root[f[lca_][]],root[u],root[v],k,,size);
            last=dis_[last];
            printf("%d\n",last);
        }
    }
    return ;
}