3545: [ONTAK2010]Peaks
Time Limit: 10 Sec Memory Limit: 128 MBSubmit: 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
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
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 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
6
1
-1
8
*/