【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。

题解

　　离散后排序，维护加边顺序，然后就是线段树合并了，权值线段树。

  1 #include<cstring>
  2 #include<cmath>
  3 #include<algorithm>
  4 #include<iostream>
  5 #include<cstdio>
  6 
  7 #define N 100007
  8 #define M 500007
  9 #define ll long long
 10 using namespace std;
 11 inline int read()
 12 {
 13     int x=0,f=1;char ch=getchar();
 14     while(ch>'9'||ch<'0'){if (ch=='-') f=-1;ch=getchar();}
 15     while(ch<='9'&&ch>='0'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
 16     return x*f;
 17 }
 18 
 19 int n,m,q,sz;
 20 int fa[N],rt[N],ans[M],disc[N],h[N];
 21 int siz[M*10],ls[M*10],rs[M*10];
 22 struct Node
 23 {
 24     int x,y,difficulty;
 25 }a[M];
 26 struct Date
 27 {
 28     int x,limit,k,id;
 29 }b[M];
 30 
 31 int find(int num)
 32 {
 33     if (fa[num]!=num) fa[num]=find(fa[num]);
 34     return fa[num];
 35 }
 36 bool cmp(Node x,Node y)
 37 {
 38     return x.difficulty<y.difficulty;
 39 }
 40 bool cmp1(Date x,Date y)
 41 {
 42     return x.limit<y.limit;
 43 }
 44 
 45 int merge(int x,int y)
 46 {
 47     if (!x)return y;
 48     if (!y)return x;
 49     if (!ls[x]&&!rs[x])
 50     {
 51         siz[x]=siz[x]+siz[y];
 52         return x;
 53     }
 54     ls[x]=merge(ls[x],ls[y]);
 55     rs[x]=merge(rs[x],rs[y]);
 56     siz[x]=siz[ls[x]]+siz[rs[x]];
 57     return x;
 58 }
 59 void ins(int &p,int l,int r,int z)
 60 {
 61     if (!p)p=++sz,siz[p]=1;
 62     if (l==r) return;
 63     int mid=(l+r)>>1;
 64     if (z<=mid)ins(ls[p],l,mid,z);
 65     else ins(rs[p],mid+1,r,z);
 66 }
 67 int query(int p,int l,int r,int rank)
 68 {
 69     if (l==r) return l;
 70     int mid=(l+r)>>1;
 71     if (rank<=siz[ls[p]])return query(ls[p],l,mid,rank);
 72     else return query(rs[p],mid+1,r,rank-siz[ls[p]]);
 73 }
 74 void solve()
 75 {
 76     int now=0;
 77     for (int i=1;i<=q;i++)
 78     {
 79         while(now<m&&a[now+1].difficulty<=b[i].limit)
 80         {
 81             int x=find(a[now+1].x),y=find(a[now+1].y);
 82             if (x!=y)
 83             {
 84                 fa[y]=x;
 85                 rt[x]=merge(rt[x],rt[y]);
 86             }
 87             now++;
 88         }
 89         int x=find(b[i].x);
 90         if (siz[rt[x]]<b[i].k) ans[b[i].id]=-1;
 91         else ans[b[i].id]=disc[query(rt[x],1,n,siz[rt[x]]-b[i].k+1)];
 92     }
 93     for (int i=1;i<=q;i++)
 94         printf("%d
",ans[i]);
 95 }
 96 int main()
 97 {
 98     freopen("fzy.in","r",stdin);
 99     freopen("fzy.out","w",stdout);
100     
101     n=read(),m=read(),q=read();
102     for (int i=1;i<=n;i++)
103         disc[i]=h[i]=read(),fa[i]=i;
104     sort(disc+1,disc+n+1);
105     for (int i=1;i<=n;i++)
106         h[i]=lower_bound(disc+1,disc+n+1,h[i])-disc;
107     for (int i=1;i<=n;i++)
108         ins(rt[i],1,n,h[i]);
109     
110     for (int i=1;i<=m;i++)
111         a[i].x=read(),a[i].y=read(),a[i].difficulty=read();
112     sort(a+1,a+m+1,cmp);
113     for (int i=1;i<=q;i++)
114         b[i].x=read(),b[i].limit=read(),b[i].k=read(),b[i].id=i;
115     sort(b+1,b+q+1,cmp1);
116     solve();
117 }