【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

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

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

Sample Output

6

1

-1

8

HINT

【数据范围】

N<=10^5, M,Q<=5*10^5，h_i,c,x<=10^9。

题解
离线处理+线段树合并。
按照边权从小到大处理，每个连通块开开一个权值线段树用于查找第k高的山峰。

代码

#include<bits/stdc++.h>
#define ll long long
#define inf 1000000000
#define mod 201314
using namespace std;
const int N=100005;
inline 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*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,m,q,sz;
int Ls[100005],f[100005],h[100005],rt[100005],ans[500005];
struct node{int a,b,w,ans,id;}a[1000005];
int ls[5000005],rs[5000005],sum[5000005];
bool cmp(node a,node b){return a.w<b.w||(a.w==b.w&&a.id<b.id);}
int find(int i){if (f[i]!=i)f[i]=find(f[i]);return f[i];}
void insert(int &k,int l,int r,int x)
{
    k=++sz;sum[k]++;
    if (l==r) return;
    int mid=(l+r)>>1;
    if (x<=mid) insert(ls[k],l,mid,x);
    else insert(rs[k],mid+1,r,x);
}
int merge(int a,int b)
{
    if (!a) return b;if (!b) return a;
    if(!ls[a]&&!rs[a])
    {
        sum[a]=sum[a]+sum[b];
        return a;
    }
    ls[a]=merge(ls[a],ls[b]);rs[a]=merge(rs[a],rs[b]);
    sum[a]=sum[ls[a]]+sum[rs[a]];
    return a;
}
int query(int k,int l,int r,int x)
{
    if (l==r) return l;
    int mid=(l+r)>>1;
    if (sum[ls[k]]>=x) return query(ls[k],l,mid,x);
    else query(rs[k],mid+1,r,x-sum[ls[k]]);
}
int main()
{
    n=read();m=read();q=read();
    for (int i=1;i<=n;i++)h[i]=read(),Ls[i]=h[i];
    sort(Ls+1,Ls+n+1);
    for (int i=1;i<=n;i++)h[i]=lower_bound(Ls+1,Ls+n+1,h[i])-Ls;
    for (int i=1;i<=n;i++)f[i]=i;
    for (int i=1;i<=m;i++)
    {
        a[i].a=read();a[i].b=read();a[i].w=read();
    }
    for (int i=m+1;i<=m+q;i++)
    {
        a[i].id=i-m;a[i].a=read();a[i].w=read();a[i].b=read();
    }
    sort(a+1,a+m+q+1,cmp);
    for (int i=1;i<=n;i++) insert(rt[i],1,n,h[i]);
    for (int i=1;i<=m+q;i++)
    {
        if (a[i].id==0)
        {
            int x=find(a[i].a),y=find(a[i].b);
            if (x!=y)
            {
                f[y]=x;
                rt[x]=merge(rt[x],rt[y]);
            }
        }
        else
        {
            int x=find(a[i].a);
            if (sum[rt[x]]<a[i].b)ans[a[i].id]=-1;
            else ans[a[i].id]=Ls[query(rt[x],1,n,sum[rt[x]]-a[i].b+1)];
        }
    }
    for (int i=1;i<=q;i++)
        printf("%d\n",ans[i]);
}