bzoj2588 Count on a tree(树上建主席树求路径第k大)

主席上树(划去。
每个点建一棵权值线段树,记录从这个点到根的路径上的点权信息。x可以和fa[x]的线段树共用很多信息,所以上主席树!那么x到y路径上的点权信息,就是tree[x]+tree[y]-tree[t]-tree[fa[t]],t表示lca(x,y)。

#include 
#include 
#include 
using namespace std;
#define N 100010
int n,m,nn,a[N],b[N],num=0,h[N],fa[N][20],dep[N],Log[N],owo=0,rt[N],ans=0;
struct edge{
    int to,next;
}data[N<<1];
struct treenode{
    int x,lc,rc;
}tree[20*N];
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;
}
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 void dfs(int x){
    for(int i=1;i<=Log[n];++i){
        if(!fa[x][i-1]) break;
        fa[x][i]=fa[fa[x][i-1]][i-1];
    }rt[x]=rt[fa[x][0]];change(rt[x],1,nn,a[x]);
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(fa[x][0]==y) continue;
        fa[y][0]=x;dep[y]=dep[x]+1;dfs(y);
    }
}
inline int lca(int x,int y){
    if(dep[x]int t=dep[x]-dep[y];
    for(int i=0;i<=Log[t];++i)
        if(t>>i&1) x=fa[x][i];
    if(x==y) return x;
    for(int i=Log[n];i>=0;--i)
        if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int query(int p1,int p2,int p3,int p4,int l,int r,int k){
    if(l==r) return l;int mid=l+r>>1;
    int x=tree[tree[p1].lc].x+tree[tree[p2].lc].x-tree[tree[p3].lc].x-tree[tree[p4].lc].x;
    if(k<=x) return query(tree[p1].lc,tree[p2].lc,tree[p3].lc,tree[p4].lc,l,mid,k);
    else return query(tree[p1].rc,tree[p2].rc,tree[p3].rc,tree[p4].rc,mid+1,r,k-x);
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();Log[0]=-1;
    for(int i=1;i<=n;++i) a[i]=b[i]=read();sort(b+1,b+n+1);nn=unique(b+1,b+n+1)-b-1;
    for(int i=1;i<=n;++i) a[i]=lower_bound(b+1,b+nn+1,a[i])-b;
    for(int i=1;i<=n;++i) Log[i]=Log[i>>1]+1;
    for(int i=1;iint x=read(),y=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;
        data[++num].to=x;data[num].next=h[y];h[y]=num;
    }dfs(1);
    while(m--){
        int x=read()^ans,y=read(),k=read(),t=lca(x,y);
        printf("%d",ans=b[query(rt[x],rt[y],rt[t],rt[fa[t][0]],1,nn,k)]);if(m) puts("");
    }return 0;
}

你可能感兴趣的:(bzoj,主席树,-----树-------)