3551: [ONTAK2010]Peaks加强版

题目链接

题目大意：同3545，强制在线

题解：kruskal重构树
板子来自hzwer
讲解可以看popoqqq

我的收获：模版++

#include <bits/stdc++.h>
using namespace std;

const int L=20;
const int N=200005;

int n,m,Q;
int ext,top;
int lastans;
int h[N>>1],z[N>>1];
int st[N],ed[N],q[N<<1],root[N<<1];
int f[N],val[N];
int dep[N],fa[N][L+2],mx[N][L+2];
bool vis[N];

vector <int> G[N];

struct data{int u,v,w;}E[N<<2];
inline bool cmp(data x,data y){return x.w<y.w;}

int Find(int x){return f[x]==x?x:f[x]=Find(f[x]);}

int search(int x,int V)
{
    for(int i=L;i>=0;i--)
        if(dep[x]>=(1<<i)&&mx[x][i]<=V) x=fa[x][i];
    return x;
}

void dfs(int x)
{
    vis[x]=1;q[++top]=x;
    for(int i=1;i<=L;i++) fa[x][i]=fa[fa[x][i-1]][i-1],mx[x][i]=max(mx[x][i-1],mx[fa[x][i-1]][i-1]);
    for(int i=0;i<G[x].size();i++){
        int v=G[x][i];
        fa[v][0]=x,dep[v]=dep[x]+1;mx[v][0]=val[x];
        dfs(v);
    }
    if(x>n) q[++top]=x;
}

struct ChairmanTree{

    #define MX N*25
    #define lson tl[x],tl[y],l,mid
    #define rson tr[x],tr[y],mid+1,r

    int siz;
    int tl[MX],tr[MX],sum[MX];

    inline void node(int &x,int w){x=++siz,sum[x]=sum[w]+1,tl[x]=tl[w],tr[x]=tr[w];}

    void insert(int x,int &y,int l,int r,int k)
    {
        node(y,x);
        if(l==r) return ;
        int mid=l+r>>1;
        if(k<=mid) insert(lson,k);
        else insert(rson,k);
    }

    int query(int x,int y,int l,int r,int k)
    {
        if(l==r) return l;
        int mid=(l+r)>>1,t=sum[tl[y]]-sum[tl[x]];
        if(t>=k) return query(lson,k);
        return query(rson,k-t);
    }

}T;

void kruskal()
{
    ext=n;
    sort(E+1,E+1+m,cmp);
    for(int i=1;i<=m;i++){
        int x=Find(E[i].u),y=Find(E[i].v);
        if(x!=y)
        {
            ext++;
            f[x]=f[y]=ext;
            val[ext]=E[i].w;
            G[ext].push_back(x),G[ext].push_back(y);
            if(ext==2*n-1) break;
        }
    }
    for(int i=1;i<=n;i++) if(!vis[i]) dfs(Find(i));
    for(int i=1;i<=top;i++)
    {
        int t=q[i];
        if(t<=n) T.insert(root[i-1],root[i],1,n,h[t]);
        else
        {
            root[i]=root[i-1];
            if(!st[t]) st[t]=i;
            else ed[t]=i;
        }
    }
}

void solve()
{
    int u,v,k;
    for(int i=1;i<=Q;i++)
    {
        scanf("%d%d%d",&u,&v,&k);
        if(lastans!=-1) u^=lastans,v^=lastans,k^=lastans;
        int t=search(u,v);
        int a=root[st[t]],b=root[ed[t]];
        if(T.sum[b]-T.sum[a]<k) lastans=-1;
        else lastans=z[T.query(a,b,1,n,T.sum[b]-T.sum[a]-k+1)];
        printf("%d\n",lastans);
    }
}

void work()
{
    kruskal();
    solve();
}

void init()
{
    scanf("%d%d%d",&n,&m,&Q);
    for(int i=1;i<=n;i++) scanf("%d",&h[i]),z[i]=h[i];
    sort(z+1,z+n+1);int cnt=unique(z+1,z+1+n)-z-1;
    for(int i=1;i<=n;i++) h[i]=lower_bound(z+1,z+1+cnt,h[i])-z;
    for(int i=1;i<=m;i++) scanf("%d%d%d",&E[i].u,&E[i].v,&E[i].w);
    for(int i=1;i<=2*n;i++) f[i]=i;
}

int main()
{   
    init();
    work();
    return 0;
}