luogu4197 Peaks (kruskal重构树+主席树)

按照边权排序建出kruskal重构树，每次就变成了先找一个权值<=x的最远的祖先，然后看这个子树的第k小。离散化一下，在dfs序上做主席树即可

而且只需要建叶节点的主席树

注意输出的是第k小点的高度值

 #include<bits/stdc++.h>
 #define pa pair<int,int>
 #define ll long long
 using namespace std;
 const int maxn=1e5+,maxm=5e5+;

 inline ll rd(){
     ll x=;char c=getchar();int neg=;
     while(c<''||c>''){if(c=='-') neg=-;c=getchar();}
     while(c>=''&&c<='') x=x*+c-'',c=getchar();
     return x*neg;
 }

 struct Edge{
     int a,b,l;
 }eg[maxm];
 pa st[maxn];
 int N,M,Q,L,h[maxn<<];
 int fa[maxn<<],source[maxn];
 int tf[maxn<<][],ch[maxn<<][],pct;
 int dfn[maxn<<][],tot;
 int root[maxn],ztr[maxn*],zch[maxn*][],zct;

 inline int getf(int x){
     return x==fa[x]?x:fa[x]=getf(fa[x]);
 }

 inline bool cmp(Edge a,Edge b){
     return a.l<b.l;
 }

 void insert(int pre,int &now,int l,int r,int x){
     now=++zct;ztr[now]=ztr[pre]+;
     if(l<r){
         int m=l+r>>;
         if(x<=m) zch[now][]=zch[pre][],insert(zch[pre][],zch[now][],l,m,x);
         else zch[now][]=zch[pre][],insert(zch[pre][],zch[now][],m+,r,x);
     }
 }
 int query(int pre,int now,int l,int r,int k){
 //    printf("%d %d %d\n",l,r,k);
     if(l==r) return l;
     int w=ztr[zch[now][]]-ztr[zch[pre][]];
     int m=l+r>>;
     if(w>=k) return query(zch[pre][],zch[now][],l,m,k);
     else return query(zch[pre][],zch[now][],m+,r,k-w);
 }

 void dfs(int x){
 //    printf("%d %d %d %d\n",x,tf[x][0],ch[x][0],ch[x][1]);
     for(int i=;tf[x][i-]&&tf[tf[x][i-]][i-];i++){
         tf[x][i]=tf[tf[x][i-]][i-];
     }
     if((!ch[x][])&&(!ch[x][])){
         dfn[x][]=dfn[x][]=++tot;
         insert(root[tot-],root[tot],,L,h[x]);
     }else{
         dfn[x][]=tot+;
         dfs(ch[x][]);dfs(ch[x][]);
         dfn[x][]=tot;
     }
 }

 int find(int x,int y){
     for(int i=;i>=;i--){
         if(tf[x][i]&&h[tf[x][i]]<=y) x=tf[x][i];
     }return x;
 }

 int main(){
     //freopen("4197.in","r",stdin);
     int i,j,k;
     N=rd(),M=rd(),Q=rd();
     for(i=;i<=N;i++) st[i]=make_pair(rd(),i);
     sort(st+,st+N+);
     for(i=,j=;i<=N;i++){
         if(st[i].first!=st[i-].first) j++;
         h[st[i].second]=j;source[j]=st[i].first;
     }L=j;
     for(i=;i<=M;i++){
         int a=rd(),b=rd(),c=rd();
         eg[i].a=a;eg[i].b=b;eg[i].l=c;
     }
     sort(eg+,eg+M+,cmp);
     for(i=;i<=N*;i++) fa[i]=i;
     pct=N;
     for(i=;i<=M;i++){
         int a=getf(eg[i].a),b=getf(eg[i].b);
         if(a==b) continue;
         fa[a]=fa[b]=++pct;h[pct]=eg[i].l;
         tf[a][]=tf[b][]=pct;
         ch[pct][]=a,ch[pct][]=b;
     }
     for(i=pct;i;i--){
         if(!dfn[i][]) dfs(i);
     }
     for(i=;i<=Q;i++){
         int a=rd(),b=rd(),c=rd();
         int x=find(a,b);
         int pr=root[dfn[x][]-],no=root[dfn[x][]];
         int w=ztr[no]-ztr[pr];
         if(w<c) printf("-1\n");
         else{
             printf("%d\n",source[query(pr,no,,L,w-c+)]);
         }
     }
     return ;
 }