bzoj3545[ONTAK2010]Peaks 线段树合并+离散

一开始以为是什么没见过的点分治姿势，后来发现不对，肯定要数据结构维护，那难道是树剖？不可能啊这怎么维护= =，在ymw大神的提醒下突然想到线段树合并= =。。
先离散，然后对每一个联通块建一颗线段树（动态开点），然后按照边的从小到大排序，询问也按照那个限制从小到大排序，然后离线，按照题目要求每次把小于当前限制的并查集合并一下，同时线段树合并，然后直接在线段树内寻找k大就可以了，如果区间内的数

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=1e5+5;
struct node
{
    int x,y,z,id;
}e[N*5],q[N*5];
int f[N],val[N],a[N],root[N];
struct tree
{
    int l,r,s;
}t[N*20];
int n,m,Q,sz,ans[N*5];
bool cmp(node a,node b)
{
    return a.z<b.z;
}
bool cmpq(node a,node b)
{
    return a.y<b.y;
}
inline void ins(int &x,int l,int r,int pos,int v)
{
    if (!x)x=++sz;
    t[x].s+=v;
    if (l==r)return;
    int mid=(l+r)>>1;
    if (pos<=mid)ins(t[x].l,l,mid,pos,v);
    else ins(t[x].r,mid+1,r,pos,v); 
}
inline int query(int x,int l,int r,int k)
{
    if (l==r)return l;
    int mid=(l+r)>>1;
    if (t[t[x].r].s>=k)return query(t[x].r,mid+1,r,k);
    else return query(t[x].l,l,mid,k-t[t[x].r].s);
}
inline int find (int x)
{
    if (f[x]==x)return x;
    return f[x]=find(f[x]);
}
inline int merge(int x,int y)
{
    if (!x||!y)return x+y;
    t[x].s+=t[y].s;
    t[x].l=merge(t[x].l,t[y].l);
    t[x].r=merge(t[x].r,t[y].r);
    return x;
}
inline void union1(int x,int y)
{
    int fx=find(x),fy=find(y);
    if (fx==fy)return;
    f[fy]=fx;
    root[fx]=merge(root[fx],root[fy]);
}
int main()
{
    scanf("%d%d%d",&n,&m,&Q);
    fo(i,1,n)
    {
        scanf("%d",&val[i]);
        a[i]=val[i];f[i]=i;
    }
    sort(a+1,a+1+n);
    int n1=unique(a+1,a+n+1)-a-1;
    fo(i,1,n)
    {
        val[i]=lower_bound(a+1,a+n1+1,val[i])-a;
        ins(root[i],1,n,val[i],1);
    }
    fo(i,1,m)
    {
        scanf("%d%d%d",&e[i].x,&e[i].y,&e[i].z);
    }
    sort(e+1,e+m+1,cmp);
    fo(i,1,Q)
    {
        scanf("%d%d%d",&q[i].x,&q[i].y,&q[i].z);
        q[i].id=i;
    }
    sort(q+1,q+Q+1,cmpq);
    int now=1;
    fo(i,1,Q)
    {
        while (now<=m&&e[now].z<=q[i].y)union1(e[now].x,e[now].y),now++; 
        ans[q[i].id]=t[root[find(q[i].x)]].s>=q[i].z?a[query(root[find(q[i].x)],1,n,q[i].z)]:-1;
    }
    fo(i,1,Q)printf("%d\n",ans[i]);

}