3545: [ONTAK2010]Peaks 线段树合并

Description
在Bytemountains有N座山峰，每座山峰有他的高度h_i。有些山峰之间有双向道路相连，共M条路径，每条路径有一个困难值，这个值越大表示越难走，现在有Q组询问，每组询问询问从点v开始只经过困难值小于等于x的路径所能到达的山峰中第k高的山峰，如果无解输出-1。

题解：

将询问离线，按照x从小到大排序，每次将小于等于x的边连上，实际上就是合并一下两个点所在的线段树，然后就可以了。在线版的还不会，以后再学。

代码：

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=100010;
const int Maxm=500010;
const int Maxq=500010;
const int inf=2147483647;
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<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}
int n,m,Q,h[Maxn];
struct lsh{int x,id;}A[Maxn];
bool cmplsh(lsh a,lsh b){return a.x<b.x;}
struct Edge{int y,next;}e[Maxm<<1];
int last[Maxn],len=0;
void ins(int x,int y){int t=++len;e[t].y=y;e[t].next=last[x];last[x]=t;}
struct Query{int v,x,k,id;}q[Maxq];
bool cmp(Query a,Query b){return a.x<b.x;}
int lc[Maxn*78],rc[Maxn*78],c[Maxn*78];
int root[Maxn],tot=0,bel[Maxn],size[Maxn];
void insert(int &u,int l,int r,int p)
{
    if(!u)u=++tot;
    c[u]++;
    if(l==r)return;
    int mid=l+r>>1;
    if(p<=mid)insert(lc[u],l,mid,p);
    else insert(rc[u],mid+1,r,p);
}
void merge(int &u1,int u2)
{
    if(!u1){u1=u2;return;}
    if(!u2)return;
    c[u1]+=c[u2];
    merge(lc[u1],lc[u2]);
    merge(rc[u1],rc[u2]);
}
int vis[Maxn],now=-1,rt;
void dfs(int x,int f)
{
    size[bel[x]=rt]++;
    vis[x]=now;
    for(int i=last[x];i;i=e[i].next)
    {
        int y=e[i].y;
        if(vis[y]==now)continue;
        dfs(y,x);
    }
}
int I;
int query(int rt,int l,int r,int k)
{
    if(l==r)return l;
    int t=c[lc[rt]],mid=l+r>>1;
    if(t>=k)return query(lc[rt],l,mid,k);
    else return query(rc[rt],mid+1,r,k-t);
}
int Ans[Maxq],ex[Maxm],ey[Maxm],ed[Maxm];
struct _Edge{int x,y,d;}E[Maxm];
bool _cmp(_Edge a,_Edge b){return a.d<b.d;}
int to[Maxn];
int main()
{
    n=read(),m=read(),Q=read();
    for(int i=1;i<=n;i++)A[i].x=read(),A[i].id=i;
    sort(A+1,A+1+n,cmplsh);
    int cnt=0;A[0].x=-1;
    for(int i=1;i<=n;i++)
    {
        if(A[i].x!=A[i-1].x)cnt++;
        h[A[i].id]=cnt;to[cnt]=A[i].x;
    }
    now++;
    for(int i=1;i<=n;i++)insert(root[i],1,1e5,h[i]),rt=i,dfs(i,-1);
    for(int i=1;i<=m;i++)E[i].x=read(),E[i].y=read(),E[i].d=read();
    sort(E+1,E+1+m,_cmp);
    for(int i=1;i<=Q;i++)q[i].v=read(),q[i].x=read(),q[i].k=read(),q[i].id=i;
    sort(q+1,q+1+Q,cmp);
    int j=0;
    for(int i=1;i<=Q;i++)
    {
        while(j<m&&E[j+1].d<=q[i].x)
        {
            j++;
            int x=E[j].x,y=E[j].y;
            if(bel[x]==bel[y])continue;
            x=bel[x],y=bel[y];
            if(size[y]>size[x])swap(x,y);
            ins(x,y);ins(y,x);
            rt=x;vis[x]=++now;dfs(y,x);
            merge(root[x],root[y]);
        }
        if(size[bel[q[i].v]]<q[i].k)Ans[q[i].id]=-1;
        else Ans[q[i].id]=query(root[bel[q[i].v]],1,1e5,size[bel[q[i].v]]-q[i].k+1);
    }
    for(int i=1;i<=Q;i++)
    {
        if(Ans[i]==-1)puts("-1");
        else printf("%d\n",to[Ans[i]]);
    }
}