bzoj3545: [ONTAK2010]Peaks

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

对于每组询问,输出一个整数表示答案。

考虑离线,将边和询问按边权排序,从小到大加入边,平衡树启发式合并维护连通块内点权

#include<cstdio>
#include<algorithm>
#include<cstdlib>
const int M=;
char buf[M+],*ptr=buf-;
inline int _int(){
int x=,c=*++ptr,f=;
while(c>||c<){if(c=='-')f=-;c=*++ptr;}
while(c>&&c<)x=x*+c-,c=*++ptr;
return x;
}
struct edge{
int s,t,l;
}e[];
bool operator<(const edge&a,const edge&b){
return a.l<b.l;
}
struct Q{
int w,mx,k,id;
}qs[];
bool operator<(const Q&a,const Q&b){
return a.mx<b.mx;
}
const int N=;
int n,m,q;
int h[N],ans[];
int ch[N][],sz[N],col[N],rt[N],rnd[N];
int stk[N],stp;
void dfs(int w){
if(!w)return;
dfs(ch[w][]);
stk[stp++]=w;
dfs(ch[w][]);
}
inline void up(int w){
sz[w]=+sz[ch[w][]]+sz[ch[w][]];
}
inline void rot(int&w,int d){
int u=ch[w][d];
ch[w][d]=ch[u][d^];
ch[u][d^]=w;
up(w);
up(w=u);
}
void ins(int&w,int x){
if(!w){
w=x;
return;
}
++sz[w];
int d=(h[w]<h[x]);
ins(ch[w][d],x);
if(rnd[ch[w][d]]>rnd[w])rot(w,d);
}
int kmax(int w,int k){
if(sz[w]<k||k<=)return -;
--k;
while(){
int s=sz[ch[w][]];
if(s==k)return h[w];
if(s<k){
k-=s+;
w=ch[w][];
}else{
w=ch[w][];
}
}
}
void merge(int x,int y){
x=col[x];y=col[y];
if(x==y)return;
if(sz[rt[x]]<sz[rt[y]]){int z=x;x=y;y=z;}
stp=;
dfs(rt[y]);
for(int i=;i<stp;i++){
int w=stk[i];
col[w]=x;
sz[w]=;
ch[w][]=ch[w][]=;
ins(rt[x],w);
}
}
int main(){
fread(buf,,M,stdin);
srand();
n=_int();m=_int();q=_int();
for(int i=;i<=n;i++){
h[i]=_int();
sz[rt[i]=col[i]=i]=;
rnd[i]=rand();
}
for(int i=;i<m;i++){
e[i].s=_int();
e[i].t=_int();
e[i].l=_int();
}
std::sort(e,e+m);
e[m].l=;
for(int i=;i<q;i++){
qs[i].w=_int();
qs[i].mx=_int();
qs[i].k=_int();
qs[i].id=i;
}
std::sort(qs,qs+q);
for(int i=,p=;i<q;i++){
int mx=qs[i].mx;
while(e[p].l<=mx){
merge(e[p].s,e[p].t);
++p;
}
ans[qs[i].id]=kmax(rt[col[qs[i].w]],qs[i].k);
}
for(int i=;i<q;i++)printf("%d\n",ans[i]);
return ;
}
上一篇:利用box-shadow绘图


下一篇:第六天 文件的基本管理和xfs文进系统备份恢复