BZOJ 3545: [ONTAK2010]Peaks 线段树合并

3545: [ONTAK2010]Peaks

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 2238  Solved: 597
[Submit][Status][Discuss]

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。

我们只要支持动态维护连通性，并且支持查询就好了

对每一座山建权值线段树

搞线段树合并就好了

空间开小RE一发  m错写成n WA一发 哈哈哈

#include<cmath>#include<ctime>#include<cstdio>#include<cstring>#include<cstdlib>#include<iostream>#include<algorithm>#include<iomanip>#include<vector>#include<string>#include<bitset>#include<queue>#include<set>#include<map>using namespace std;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<<1)+(x<<3)+ch-'0';ch=getchar();}return x*f;}void print(int x){if(x<0)x=-x,putchar('-');if(x>=10)print(x/10);putchar(x%10+'0');}const int N=100100,M=500100;int n,m,Q,maxn,tot;struct question{int u,x,k,pos;friend bool operator <(const question &x,const question &y){return x.x<y.x;}}q[M];struct EDGE{int fr,to,val;friend bool operator <(const EDGE &x,const EDGE &y){return x.val<y.val;}}e[M];int R[N],V[N],root[N],fa[N],ans[M];inline int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}struct seg_tree{int ls,rs,sum;}str[N*20];void modify(int &k,int l,int r,int val){if(!k)k=++tot;str[k].sum++;if(l==r)return ;int mid=(l+r)>>1;if(val<=mid)modify(str[k].ls,l,mid,val);else modify(str[k].rs,mid+1,r,val);}int merger(int x,int y){if(!x)return y;if(!y)return x;if(!str[x].ls&&!str[x].rs){str[x].sum=str[x].sum+str[y].sum;return x;}str[x].ls=merger(str[x].ls,str[y].ls);str[x].rs=merger(str[x].rs,str[y].rs);str[x].sum=str[str[x].ls].sum+str[str[x].rs].sum;return x;}int query(int k,int l,int r,int rk){if(l==r)return l;int mid=(l+r)>>1;if(str[str[k].ls].sum>=rk)return query(str[k].ls,l,mid,rk);return query(str[k].rs,mid+1,r,rk-str[str[k].ls].sum);}void solve(){register int i,now=1,u,v,x,y;for(i=1;i<=n;++i)fa[i]=i,modify(root[i],1,maxn,V[i]);for(i=1;i<=Q;++i){while(now<=m&&e[now].val<=q[i].x){u=e[now].fr,v=e[now].to;x=find(u);y=find(v);if(x^y){root[x]=merger(root[x],root[y]);fa[y]=x;}now++;}u=q[i].u;x=find(u);if(str[root[x]].sum<q[i].k)ans[q[i].pos]=-1;else ans[q[i].pos]=query(root[x],1,maxn,str[root[x]].sum-q[i].k+1);}}int main(){n=read();m=read();Q=read();register int i,k=0;for(i=1;i<=n;++i)q[i].x=read(),q[i].k=i;sort(q+1,q+1+n);k++;R[1]=q[1].x;V[q[1].k]=1;for(i=2;i<=n;++i){if(q[i].x^R[k])k++,R[k]=q[i].x;V[q[i].k]=k;}maxn=k;for(i=1;i<=m;++i)e[i].fr=read(),e[i].to=read(),e[i].val=read();sort(e+1,e+1+m);for(i=1;i<=Q;++i)q[i].u=read(),q[i].x=read(),q[i].k=read(),q[i].pos=i;sort(q+1,q+1+Q);solve();for(i=1;i<=Q;++i){print(ans[i]!=-1?R[ans[i]]:ans[i]);puts("");}return 0;}/*10 11 41 2 3 4 5 6 7 8 9 101 4 42 5 39 8 27 8 107 1 46 7 16 4 82 1 510 8 103 4 73 4 61 5 21 5 61 5 88 9 261-18*/