BZOJ3551 : [ONTAK2010]Peaks加强版

首先强制在线的话，肯定是不能再离线排序+平衡树启发式合并了。

这回要用的是线段树合并，每次把两棵线段树合并，总复杂度为$O(nlog n)$

预处理：

把边按权值从小到大排序，依次加边，

对于边(x,y)，权值为z，如果x和y已经在一个联通块里就无视掉

假设x块大小小于等于y块大小

将x,y两块的线段树合并，设合并后线段树根为r，并在y所在块根节点处root表后面加入一个(r,z)

然后把x块内所有点的top表后面加入一个(top[y],z)

这里启发式合并的总复杂度也为$O(nlog n)$

查询从v出发走权值不超过x所到达的点中第k大：

先在v的top表里二分出最后面的权值不超过x的根节点r

再在r的root表里二分出最后面的权值不超过x的线段树根节点t

最后在t树中查询第k大

所以查询一次的复杂度为$O(log n)$

所以总复杂度为$O((n+q)log n)$

（吐槽：$O(nlog n)$的线段树合并不知道为什么比我之前$O(nlog^2n)$的平衡树启发式合并还要慢1S）

#include<cstdio>
#include<algorithm>
#define N 100010
#define M 3500000
#define T 1500000
using namespace std;
inline void read(int&a){char c;while(!(((c=getchar())>='0')&&(c<='9')));a=c-'0';while(((c=getchar())>='0')&&(c<='9'))(a*=10)+=c-'0';}
struct edge{int a,b,c;}e[500010];
inline bool cmp(edge a,edge b){return a.c<b.c;}
int n,m,q,i,j,et,x,y,k,last;
int h[N],b[N];
inline int lower(int x){
  int l=1,r=n,mid,t;
  while(l<=r)if(b[mid=(l+r)>>1]<x)l=mid+1;else r=(t=mid)-1;
  return t;
}
int l[M],r[M],val[M],tot;
int merge(int x,int y,int a,int b){
  if(!x)return y;
  if(!y)return x;
  int z=++tot;
  if(a==b){
    val[z]=val[x]+val[y];
    return z;
  }
  int mid=(a+b)>>1;
  l[z]=merge(l[x],l[y],a,mid);
  r[z]=merge(r[x],r[y],mid+1,b);
  val[z]=val[l[z]]+val[r[z]];
  return z;
}
int build(int a,int b,int c){
  int x=++tot;
  val[x]=1;
  if(a==b)return x;
  int mid=(a+b)>>1;
  if(c<=mid)l[x]=build(a,mid,c);else r[x]=build(mid+1,b,c);
  return x;
}
int kth(int x,int k){
  if(k>val[x])return 0;
  int a=1,b=n,mid,t;
  while(1){
    if(a==b)return a;
    mid=(a+b)>>1,t=val[r[x]];
    if(k<=t)x=r[x],a=mid+1;else k-=t,x=l[x],b=mid;
  }
}
int size[N],g[N],nxt[N<<1],v[N<<1],ed;
int top_st[N],top_en[N],top_nxt[T],top_v[T],top_w[T],top_list[T],top_bg[N],top_cnt[N],td;
int root_st[N],root_en[N],root_nxt[N<<1],root_v[N<<1],root_w[N<<1],root_list[N<<1],root_bg[N],root_cnt[N],rd;
inline void addedge(int x,int y){v[++ed]=y;nxt[ed]=g[x];g[x]=ed;}
inline void addtop(int x,int y,int z){top_v[++td]=y;top_w[td]=z;top_nxt[top_en[x]]=td;top_en[x]=td;}
inline void addroot(int x,int y,int z){root_v[++rd]=y;root_w[rd]=z;root_nxt[root_en[x]]=rd;root_en[x]=rd;}
void go(int x,int pre,int y,int t){
  addtop(x,y,t);
  for(int i=g[x];i;i=nxt[i])if(v[i]!=pre)go(v[i],x,y,t);
}
inline void link(int x,int y,int t){
  if(top_v[top_en[x]]==top_v[top_en[y]])return;
  if(size[top_v[top_en[x]]]>size[top_v[top_en[y]]])swap(x,y);
  addroot(top_v[top_en[y]],merge(root_v[root_en[top_v[top_en[x]]]],root_v[root_en[top_v[top_en[y]]]],1,n),t);
  size[top_v[top_en[y]]]+=size[top_v[top_en[x]]];
  go(x,0,top_v[top_en[y]],t);
  addedge(x,y);
  addedge(y,x);
}
inline int gettop(int x,int y){
  int l=top_bg[x],r=l+top_cnt[x]-1,mid,t;
  while(l<=r)if(top_w[top_list[mid=(l+r)>>1]]<=y)t=top_v[top_list[mid]],l=mid+1;else r=mid-1;
  return t;
}
inline int getroot(int x,int y){
  int l=root_bg[x],r=l+root_cnt[x]-1,mid,t;
  while(l<=r)if(root_w[root_list[mid=(l+r)>>1]]<=y)t=root_v[root_list[mid]],l=mid+1;else r=mid-1;
  return t;
}
int main(){
  read(n);read(m);read(q);
  for(i=1;i<=n;i++)read(h[i]),b[i]=h[i];b[0]=-1;
  sort(b+1,b+n+1);
  for(i=1;i<=n;i++)root_v[i]=build(1,n,lower(h[i])),top_st[i]=top_en[i]=top_v[i]=root_st[i]=root_en[i]=i,top_w[i]=root_w[i]=-1,size[i]=1;
  td=rd=n;
  for(i=1;i<=m;i++)read(e[i].a),read(e[i].b),read(e[i].c);
  sort(e+1,e+m+1,cmp);
  for(i=1;i<=m;i++)link(e[i].a,e[i].b,e[i].c);
  td=rd=0;
  for(i=1;i<=n;i++){
    for(top_bg[i]=td+1,j=top_st[i];j;j=top_nxt[j])++top_cnt[i],top_list[++td]=j;
    for(root_bg[i]=rd+1,j=root_st[i];j;j=root_nxt[j])++root_cnt[i],root_list[++rd]=j;
  }
  while(q--){
    read(x);read(y);read(k);
    if(~last)x^=last,y^=last,k^=last;
    printf("%d
",last=b[kth(getroot(gettop(x,y),y),k)]);
  }
  return 0;
}