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 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
*/