3123: [Sdoi2013]森林
Time Limit: 20 Sec Memory Limit: 512 MBSubmit: 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
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
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[25000005]; struct EdgeType { int v,next; }; struct EdgeType edge[maxn<<2]; int if_z,t,n,m,size,dis[maxn],head[maxn]; int cnt,f[maxn][22],ft[maxn],deep[maxn]; int num[maxn],dis_[maxn],last,tot,root[maxn]; char Cget; inline void in(int &now) { now=0,if_z=1,Cget=getchar(); while(Cget>'9'||Cget<'0') { if(Cget=='-') if_z=-1; Cget=getchar(); } while(Cget>='0'&&Cget<='9') { now=now*10+Cget-'0'; 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+1;tree[now].r=tree[pre].r;tree[now].l=tree[pre].l; if(l==r) return ; int mid=(l+r)>>1; if(to<=mid) { tree_add(tree[pre].l,tree[now].l,l,mid,to); } else { tree_add(tree[pre].r,tree[now].r,mid+1,r,to); } } void search(int now,int pre) { f[now][0]=pre,deep[now]=deep[pre]+1; for(int i=1;i<=20;i++) f[now][i]=f[f[now][i-1]][i-1]; int pos=lower_bound(dis_+1,dis_+size+1,dis[now])-dis_; tree_add(root[pre],root[now],1,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)>>1; tree_build(tree[now].l,l,mid); tree_build(tree[now].r,mid+1,r); } int get_same(int u,int x) { for(int i=0;i<=20;i++) if(x&(1<<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=20;i>=0;i--) { if(f[x][i]!=f[y][i]) { x=f[x][i]; y=f[y][i]; } } return f[x][0]; } 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)>>1; if(k>pos) return tree_query(tree[pre].r,tree[fpre].r,tree[now].r,tree[now_].r,k-pos,mid+1,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=1;i<=n;i++) { in(dis[i]); num[i]=1; dis_[i]=dis[i],ft[i]=i; } sort(dis_+1,dis_+n+1); size=unique(dis_+1,dis_+n+1)-dis_-1; 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[0],1,size); for(int i=1;i<=n;i++) { if(ft[i]==i) search(i,0); } char ch[10]; while(t--) { scanf("%s",ch); if(ch[0]=='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_][0]],root[u],root[v],k,1,size); last=dis_[last]; printf("%d ",last); } } return 0; }