Description
小Z有一片森林,含有N个节点,每个节点上都有一个非负整数作为权值。初始的时候,森林中有M条边。
小Z希望执行T个操作,操作有两类:
1. Q x y k
查询点x到点y路径上所有的权值中,第k小的权值是多少。此操作保证点x和点y连通,同时这两个节点的路径上至少有k个点。
2. L x y
在点x和点y之间连接一条边。保证完成此操作后,仍然是一片森林。
为了体现程序的在线性,我们把输入数据进行了加密。设lastans为程序上一次输出的结果,初始的时候lastans为0。
对于一个输入的操作Q x y k
,其真实操作为Q x^lastans y^lastans k^lastans
。
对于一个输入的操作L x y
,其真实操作为L x^lastans y^lastans
。其中^运算符表示异或,等价于pascal中的xor运算符。
请写一个程序來帮助小Z完成这些操作。
对于所有的数据,n,m,T<= 8*10^48∗104 .
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
8 4 8
1 1 2 2 3 3 4 4
4 7
1 8
2 4
2 1
Q 8 7 3
Q 10 0 0
L 5 4
L 3 2
Q 9 2 5
Sample Output
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。之后的操作类似。
题解Here!
首先看到第k小,立马想到主席树!
于是树上第k小就这么愉快的解决了: 对 u,v,LCA(u,v),fa[ LCA(u,v) ] 整体处理——
int query(int u,int v,int f,int gf,int k,int l,int r){
if(l==r)return l;
int mid=l+r>>1,t=a[a[u].l].sum+a[a[v].l].sum-a[a[f].l].sum-a[a[gf].l].sum;
if(k<=t)return query(a[u].l,a[v].l,a[f].l,a[gf].l,k,l,mid);
else return query(a[u].r,a[v].r,a[f].r,a[gf].r,k-t,mid+1,r);
}
插入主席树时,用父节点的主席树更新此节点的主席树——
void insert(int k,int l,int r,int &rt,int fa){
int mid;
rt=++size;a[rt]=a[fa];
if(l==k&&k==r){
a[rt].sum++;
return;
}
mid=l+r>>1;
if(k<=mid)insert(k,l,mid,a[rt].l,a[fa].l);
else insert(k,mid+1,r,a[rt].r,a[fa].r);
pushup(rt);
}
但是有个连边操作很烦人。。。
两个节点连边,就是两棵子树的合并。
主席树当然可以做到 O(logn) 时间内启发式合并,但是LCA怎么办?又不能启发式。。。
等等,不能启发式就暴力 dfs 啊!
时限 2s ,显然可以卡一卡。。。
由于有修改,LCA当然选倍增。
于是,一个带倍增处理的暴力 dfs 破空而出——
void buildtree(int rt,int fa,int ancestry){
f[rt][0]=fa;
for(int i=1;i<=19;i++)f[rt][i]=f[f[rt][i-1]][i-1];
size[ancestry]++;
deep[rt]=deep[fa]+1;
father[rt]=fa;
int x=lower_bound(num+1,num+p+1,val[rt])-num;
ST::insert(x,1,p,root[rt],root[fa]);
for(int i=head[rt];i;i=a[i].next){
int will=a[i].to;
if(will==fa)continue;
buildtree(will,rt,ancestry);
}
}
然后那个倍增 LCA 就不多说了。
注意:
1. 记得离散化!记得离散化!记得离散化!
2. 那个 testcase 是测试点编号!不是组数!也就是为了方便我们这些蒟蒻打打暴力。。。
3. 记得优化一下常数,什么 inline 啦,快速读入啦乱搞一波。。。
附代码:
#include<iostream>
#include<algorithm>
#include<cstdio>
#define MAXN 80010
using namespace std;
int n,m,q,p,c=1;
int val[MAXN],num[MAXN],root[MAXN];
int head[MAXN],deep[MAXN],size[MAXN],f[MAXN][20],father[MAXN];
struct Edge{
int next,to;
}a[MAXN<<2];
inline int read(){
int date=0,w=1;char c=0;
while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();}
return date*w;
}
namespace ST{
int size=0;
struct Segment{
int l,r,sum;
}a[MAXN*400];
inline void pushup(int rt){
a[rt].sum=a[a[rt].l].sum+a[a[rt].r].sum;
}
inline void buildtree(){
a[0].l=a[0].r=a[0].sum=root[0]=0;
}
void insert(int k,int l,int r,int &rt,int fa){
int mid;
rt=++size;a[rt]=a[fa];
if(l==k&&k==r){
a[rt].sum++;
return;
}
mid=l+r>>1;
if(k<=mid)insert(k,l,mid,a[rt].l,a[fa].l);
else insert(k,mid+1,r,a[rt].r,a[fa].r);
pushup(rt);
}
int query(int u,int v,int f,int gf,int k,int l,int r){
if(l==r)return l;
int mid=l+r>>1,t=a[a[u].l].sum+a[a[v].l].sum-a[a[f].l].sum-a[a[gf].l].sum;
if(k<=t)return query(a[u].l,a[v].l,a[f].l,a[gf].l,k,l,mid);
else return query(a[u].r,a[v].r,a[f].r,a[gf].r,k-t,mid+1,r);
}
}
inline int find(int x){return father[x]==x?x:father[x]=find(father[x]);}
inline void add(int x,int y){
a[c].to=y;a[c].next=head[x];head[x]=c++;
a[c].to=x;a[c].next=head[y];head[y]=c++;
}
void buildtree(int rt,int fa,int ancestry){
f[rt][0]=fa;
for(int i=1;i<=19;i++)f[rt][i]=f[f[rt][i-1]][i-1];
size[ancestry]++;
deep[rt]=deep[fa]+1;
father[rt]=fa;
int x=lower_bound(num+1,num+p+1,val[rt])-num;
ST::insert(x,1,p,root[rt],root[fa]);
for(int i=head[rt];i;i=a[i].next){
int will=a[i].to;
if(will==fa)continue;
buildtree(will,rt,ancestry);
}
}
int LCA(int x,int y){
if(deep[x]<deep[y])swap(x,y);
for(int i=19;i>=0;i--)if(deep[f[x][i]]>=deep[y])x=f[x][i];
if(x==y)return x;
for(int i=19;i>=0;i--)if(f[x][i]!=f[y][i]){x=f[x][i];y=f[y][i];}
return f[x][0];
}
void work(){
char ch[2];
int x,y,k,last=0;
while(q--){
scanf("%s",ch);x=read()^last;y=read()^last;
if(ch[0]=='Q'){
k=read()^last;
int fa=LCA(x,y);
last=num[ST::query(root[x],root[y],root[fa],root[f[fa][0]],k,1,p)];
printf("%d\n",last);
}
if(ch[0]=='L'){
add(x,y);
int fx=find(x),fy=find(y);
if(size[fx]<size[fy]){
swap(x,y);
swap(fx,fy);
}
buildtree(y,x,fx);
}
}
}
void init(){
int x,y,cases=read();
n=read();m=read();q=read();
for(int i=1;i<=n;i++){
num[i]=val[i]=read();
father[i]=i;
}
for(int i=1;i<=m;i++){
x=read();y=read();
add(x,y);
}
sort(num+1,num+n+1);
p=unique(num+1,num+n+1)-num-1;
ST::buildtree();
for(int i=1;i<=n;i++)if(!deep[i]){buildtree(i,0,i);father[i]=i;}
}
int main(){
init();
work();
return 0;
}