Description
在Bytemountains有N座山峰,每座山峰有他的高度h_i。有些山峰之间有双向道路相连,共M条路径,每条路径有一个困难值,这个值越大表示越难走,现在有Q组询问,每组询问询问从点v开始只经过困难值小于等于x的路径所能到达的山峰中第k高的山峰,如果无解输出-1。
Input
第一行三个数N,M,Q。
第二行N个数,第i个数为h_i
接下来M行,每行3个数a b c,表示从a到b有一条困难值为c的双向路径。
接下来Q行,每行三个数v x k,表示一组询问。
Output
对于每组询问,输出一个整数表示答案。
Sample Input
10 11 4
1 2 3 4 5 6 7 8 9 10
1 4 4
2 5 3
9 8 2
7 8 10
7 1 4
6 7 1
6 4 8
2 1 5
10 8 10
3 4 7
3 4 6
1 5 2
1 5 6
1 5 8
8 9 2
1 2 3 4 5 6 7 8 9 10
1 4 4
2 5 3
9 8 2
7 8 10
7 1 4
6 7 1
6 4 8
2 1 5
10 8 10
3 4 7
3 4 6
1 5 2
1 5 6
1 5 8
8 9 2
Sample Output
6
1
-1
8
1
-1
8
HINT
【数据范围】
N<=10^5, M,Q<=5*10^5,h_i,c,x<=10^9。
Source
By Sbullet
splay+启发式合并~
离线处理,把询问排序,路径也排序,并查集记录根节点位置,每次按照路径把对应的splay tree合并,求第k大。
注意:1.这里求的是第k小,要反着求。
2.样例里面山的序号和值相同,我觉得是故意的。
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;int n,m,q,a[100001],c[5000001][2],f[100001],fa[5000001],siz[5000001],root[1000001],ans[500001];struct node{int x,y,val;
}ro[500001];struct node1{int x,val,k,id;
}que[500001];int read()
{int x=0,f=1;char ch=getchar();while(ch<'0' || ch>'9') {if(ch=='-') f=-1;ch=getchar();}while(ch>='0' && ch<='9') {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}return x*f;
}bool operator < (node u,node v)
{return u.val<v.val;
}bool operator < (node1 u,node1 v)
{return u.val<v.val;
}int findd(int u)
{return f[u]==u ? u:f[u]=findd(f[u]);
}void update(int u)
{if(!u) return;siz[u]=siz[c[u][0]]+siz[c[u][1]]+1;
}void rotate(int x,int &k)
{int y=fa[x],z=fa[y],l,r;if(c[y][0]==x) l=0;else l=1;r=l^1;if(k==y) k=x;else if(c[z][0]==y) c[z][0]=x;else c[z][1]=x;fa[x]=z;fa[y]=x;c[y][l]=c[x][r];fa[c[x][r]]=y;c[x][r]=y;update(y);update(x);
}void splay(int x,int &k)
{while(x!=k){int y=fa[x],z=fa[y];if(y!=k){if(c[y][0]==x ^ c[z][0]==y) rotate(x,k);else rotate(y,k);}rotate(x,k);}
}void add(int num,int u,int v)
{if(!root[v]){root[v]=num;fa[num]=c[num][0]=c[num][1]=0;siz[num]=1;return;}int now=root[v],faa;while(1){faa=now;now=c[now][u>a[now]];if(!now){siz[num]=1;c[num][0]=c[num][1]=0;c[faa][u>a[faa]]=num;fa[num]=faa;update(num);update(faa);splay(num,root[v]);return;}}
}void del(int u,int v)
{if(c[u][0]) del(c[u][0],v);if(c[u][1]) del(c[u][1],v);add(u,a[u],v);
}int cal(int k,int v)
{while(1){if(siz[c[k][0]]>=v) k=c[k][0];else if(siz[c[k][0]]+1<v) v-=siz[c[k][0]]+1,k=c[k][1];else return k;}
}int main()
{n=read();m=read();q=read();for(int i=1;i<=n;i++) a[i]=read(),f[i]=i,siz[i]=1,root[i]=i;for(int i=1;i<=m;i++) ro[i].x=read(),ro[i].y=read(),ro[i].val=read();for(int i=1;i<=q;i++) que[i].x=read(),que[i].val=read(),que[i].k=read(),que[i].id=i;sort(ro+1,ro+m+1);sort(que+1,que+q+1);for(int k=1,now=1;k<=q;k++){if(que[k].val>que[k-1].val)for(;ro[now].val<=que[k].val && now<=m;now++){int x=findd(ro[now].x),y=findd(ro[now].y);if(x==y) continue;if(siz[root[x]]>siz[root[y]]) swap(x,y);del(root[x],y);f[x]=y;}int x=findd(que[k].x);if(siz[root[x]]<que[k].k) ans[que[k].id]=-1;else ans[que[k].id]=a[cal(root[x],siz[root[x]]-que[k].k+1)];}for(int i=1;i<=q;i++) printf("%d\n",ans[i]);return 0;
}