bzoj3545 [ONTAK2010]Peaks（离线+kruskal+主席树+线段树合并）

首先我们可以离线做。把询问按x排序，给每个点建一棵权值线段树。然后一边跑mst，一边维护线段树。每次合并两个点的时候，就要合并两棵线段树。合并的时候要小心因为合并叶子时不能pushup，所以还是每次相加比较稳妥。然后合并的总复杂度应该就是合并时删掉的节点数，是O(nlogn)的，可以接受。所以总复杂度O(nlogn+mlogm)

#include <bits/stdc++.h>using namespace std;#define ll long long#define inf 0x3f3f3f3f#define N 100010#define M 500010inline 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*10+ch-'0',ch=getchar();    return x*f;}int n,nn,m,qq,a[N],b[N],num=0,h[N],ans[M],owo=0,fa[N],rt[N];struct Data{    int x,y,val;}e[M];inline bool cmp(Data x,Data y){return x.val<y.val;}struct que{    int v,x,k,id;}q[M];inline bool cmp1(que x,que y){return x.x<y.x;}struct node{    int x,lc,rc;}tree[N*20];inline int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}inline void change(int &p,int l,int r,int x){    tree[++owo]=tree[p];p=owo;tree[p].x++;    if(l==r) return;int mid=l+r>>1;    if(x<=mid) change(tree[p].lc,l,mid,x);    else change(tree[p].rc,mid+1,r,x);}inline int merge(int p1,int p2){    if(!p1||!p2) return p1+p2;    tree[p1].lc=merge(tree[p1].lc,tree[p2].lc);    tree[p1].rc=merge(tree[p1].rc,tree[p2].rc);    tree[p1].x+=tree[p2].x;return p1;}inline int ask(int p,int l,int r,int k){    if(l==r) return l;int mid=l+r>>1;    if(k<=tree[tree[p].lc].x) return ask(tree[p].lc,l,mid,k);    else return ask(tree[p].rc,mid+1,r,k-tree[tree[p].lc].x);}int main(){//  freopen("a.in","r",stdin);    nn=read();m=read();qq=read();    for(int i=1;i<=nn;++i) a[i]=b[i]=read(),fa[i]=i;    sort(b+1,b+nn+1);n=unique(b+1,b+nn+1)-b-1;    for(int i=1;i<=nn;++i) a[i]=lower_bound(b+1,b+n+1,a[i])-b;    for(int i=1;i<=nn;++i) change(rt[i],1,n,a[i]);    for(int i=1;i<=m;++i) e[i].x=read(),e[i].y=read(),e[i].val=read();    for(int i=1;i<=qq;++i) q[i].v=read(),q[i].x=read(),q[i].k=read(),q[i].id=i;    sort(e+1,e+m+1,cmp);sort(q+1,q+qq+1,cmp1);int now=1;    for(int i=1;i<=qq;++i){        while(now<=m&&e[now].val<=q[i].x){            int x=find(e[now].x),y=find(e[now].y);            if(x!=y){                fa[x]=y;rt[y]=merge(rt[x],rt[y]);            }now++;        }if(tree[rt[find(q[i].v)]].x<q[i].k) ans[q[i].id]=-1;        else ans[q[i].id]=b[ask(rt[fa[q[i].v]],1,n,tree[rt[fa[q[i].v]]].x-q[i].k+1)];    }for(int i=1;i<=qq;++i) printf("%d\n",ans[i]);    return 0;}