bzoj 3551 kruskal重构树dfs序上的主席树

强制在线

kruskal重构树，每两点间的最大边权即为其lca的点权。

倍增找，dfs序对应区间搞主席树

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#define N 100005
#define M 500005
using namespace std;
 
int l[2*N],r[2*N],cnt,num_cnt,val[2*N],num[2*N],ans;
int sum[25*N],lon[25*N],ron[25*N];
int be[2*N],fa[2*N][21],dis[2*N][21],n,m,q,sz,oo,root[2*N];
int e=1,head[2*N];
bool vis[2*N];
struct edge{
    int u,v,w,next;
}ed[M],a[M];
bool cmp1(edge a,edge b){return a.w<b.w;}
 
void add(int u,int v,int w){
    ed[e].u=u; ed[e].v=v; ed[e].w=w;
    ed[e].next=head[u]; head[u]=e++;
}
int find(int x){
    if(x==be[x])return x;
    be[x]=find(be[x]);
    return be[x];
}
void Kruskal(){
    sort(a+1,a+m+1,cmp1);
    for(int i=1;i<=m;++i){
        int u=a[i].u,v=a[i].v;
        u=find(u); v=find(v);
        if(u!=v){
            be[u]=be[v]=++n;
            fa[u][0]=fa[v][0]=n;
            dis[u][0]=dis[v][0]=a[i].w;
            add(n,u,a[i].w); add(n,v,a[i].w);
        }
    }
}
void insert(int p,int &rt,int l,int r,int x){
    rt=++sz;
    sum[rt]=sum[p]+1;
    if(l==r) return;
    lon[rt]=lon[p]; ron[rt]=ron[p];
    int mid=(l+r)>>1;
    if(x<=mid) insert(lon[p],lon[rt],l,mid,x);
    else insert(ron[p],ron[rt],mid+1,r,x);
}
void dfs(int x){
	for(int i=1;i<=20;i++){
        fa[x][i]=fa[fa[x][i-1]][i-1];
        dis[x][i]=max(dis[x][i-1],dis[fa[x][i-1]][i-1]);
    }
    l[x]=++cnt; 
    if(x<=oo) insert(root[cnt-1],root[cnt],1,num_cnt,val[x]);
    else root[cnt]=root[cnt-1];
    for(int i=head[x];i;i=ed[i].next)
           dfs(ed[i].v);
    r[x]=cnt;
}
int query(int L,int R,int k){
    L=root[L]; R=root[R];
    if(k>sum[R]-sum[L]) return -1;
    int x=1,y=num_cnt;
    while(x<y){
        int mid=(x+y)>>1;
        int tmp=sum[ron[R]]-sum[ron[L]];
        if(tmp>=k){x=mid+1;L=ron[L];R=ron[R];}
        else{k-=tmp;y=mid;L=lon[L];R=lon[R];}
    }
    return num[x];
}
void print(int x,int l,int r){
    if(!x) return;
    printf("x==%d  l==%d  r==%d  sum==%d
",x,l,r,sum[x]);
    int mid=(l+r)>>1;
    print(lon[x],l,mid);
    print(ron[x],mid+1,r);
}
int main()
{
	//freopen("3545.in","r",stdin);
	//freopen("3545.out","w",stdout);
    int u,v,w,x,k;
    scanf("%d%d%d",&n,&m,&q); oo=n;
    for(int i=1;i<=n;i++){
        scanf("%d",&val[i]);
        num[i]=val[i];
    }
    sort(num+1,num+n+1);
    num_cnt=unique(num+1,num+n+1)-num-1;
    for(int i=1;i<=n;i++)
        val[i]=lower_bound(num+1,num+num_cnt+1,val[i])-num;
    for(int i=1;i<=m;i++)
        scanf("%d%d%d",&a[i].u,&a[i].v,&a[i].w);
    for(int i=1;i<=2*n;++i)be[i]=i;
    Kruskal();
    memset(vis,0,sizeof vis);
    for(int i=n;i>=1;--i)if(!l[i])dfs(i);
    while(q--){
        scanf("%d%d%d",&v,&x,&k);
        if(ans!=-1){v^=ans;x^=ans;k^=ans;}
        for(int i=20;~i;i--)
            if(dis[v][i]<=x&&fa[v][i])
                v=fa[v][i];
        ans=query(l[v]-1,r[v],k);
        printf("%d
",ans);
    }
    return 0;
}