bzoj3551 [ONTAK2010]Peaks加强版(kruskal重构树+dfs序+主席树+树上倍增)

题意同bzoj3545,只不过强制在线了。那强制在线了怎么搞呢x,有一个神奇的东西,kruskal重构树,张这个样子:传送门。他有很多优美的性质:首先,他一定是一颗二叉树,然后所有叶子节点一定是原图中的点,从叶子节点往跟走,点权一定是单调不降的。如果原图中有n个点,则这棵树一定有n+n-1个点(因为你需要合并n-1次,也就新建了n-1个点)。然后对于这道题而言,你可以从v开始在树上倍增找到第一个小于等于x的点,他的子树就是v能到达的所有点。这样我们就可以用dfs序+主席树求第k小啦!
离散化之前之后的n和nn傻傻分不清楚,RE到崩溃。

#include 
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100010
#define M 500010
inline 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<<1],b[N],f[N<<1],rt[N<<1],tot=0,son[N<<1][2],owo=0,ans=0;
int in[N<<1],out[N<<1],dfn=0,Log[N<<1],fa[N<<1][20],mx[N<<1][20],dep[N<<1];
struct Edge{
    int x,y,val;
}e[M];
inline bool cmp(Edge x,Edge y){return x.valstruct node{
    int lc,rc,x;
}tr[N*20];
inline int find(int x){return x==f[x]?x:f[x]=find(f[x]);}
inline void change(int &p,int l,int r,int x){
    tr[++owo]=tr[p];p=owo;tr[p].x++;
    if(l==r) return;int mid=l+r>>1;
    if(x<=mid) change(tr[p].lc,l,mid,x);
    else change(tr[p].rc,mid+1,r,x);
}
void dfs(int x){
    in[x]=++dfn;rt[dfn]=rt[dfn-1];
    for(int i=1;i<=nn<<1;++i){
        if(!fa[x][i-1]) break;
        fa[x][i]=fa[fa[x][i-1]][i-1];
    }if(x<=nn) change(rt[dfn],1,n,a[x]);
    else for(int i=0;i<2;++i){
        int y=son[x][i];dep[y]=dep[x]+1;dfs(y);
    }out[x]=dfn;
}
inline int ask(int p1,int p2,int l,int r,int x){
    if(l==r) return l;int mid=l+r>>1;
    int sz=tr[tr[p2].lc].x-tr[tr[p1].lc].x;
    if(x<=sz) return ask(tr[p1].lc,tr[p2].lc,l,mid,x);
    else return ask(tr[p1].rc,tr[p2].rc,mid+1,r,x-sz);
}
inline int find(int x,int val){
    for(int i=19;i>=0;--i){
        if(!fa[x][i]) continue;
        if(a[fa[x][i]]<=val) x=fa[x][i];
    }return 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();
    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<=m;++i) e[i].x=read(),e[i].y=read(),e[i].val=read();
    sort(e+1,e+m+1,cmp);Log[0]=-1;tot=nn;
    for(int i=1;i<=nn<<1;++i) f[i]=i,Log[i]=Log[i>>1]+1;
    for(int i=1;i<=m;++i){
        int xx=find(e[i].x),yy=find(e[i].y);
        if(xx==yy) continue;f[xx]=++tot;f[yy]=tot;
        fa[xx][0]=tot;fa[yy][0]=tot;
        son[tot][0]=xx;son[tot][1]=yy;a[tot]=e[i].val;
    }dfs(tot);
    while(qq--){
        int v=read()^ans,x=read()^ans,k=read()^ans;
        int y=find(v,x),cnt=tr[rt[out[y]]].x-tr[rt[in[y]-1]].x;
        if(cntputs("-1");ans=0;continue;}
        printf("%d\n",ans=b[ask(rt[in[y]-1],rt[out[y]],1,n,cnt-k+1)]);
    }return 0;
}

你可能感兴趣的:(-----树-------,bzoj,最小生成树,树上倍增,主席树)