BZOJ2588 主席树

来源：互联网发布：软件销售企业财务处理编辑：程序博客网时间：2024/05/24 06:51

几乎没怎么写过主席树今天看着hzwer的写了写各种RE
对于询问(u,v,k)
a=u,b=v,c=LCA(u,v),d=father(lca)
在主席树上二分s=sum[ls[a]]+sum[ls[b]]−sum[ls[c]]−sum[ls[d]];
s≤k→a,b,c,d=ls(a),ls(b),ls(c),ls(d)
s>k→a,b,c,d=rs(a),rs(b),rs(c),rs(d)

#include<bits/stdc++.h>#define hash _hashusing namespace std;const int N=2e5+5;const int M=4e6+5;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;}struct Edge{    int v,nxt;}e[N*2];int ls[M],rs[M],sum[M],dep[N],f[N][18],head[N],bin[18]={1};int n,m,tot,cnt,sz,lastans;int v[N],tmp[N],hash[N],root[N];inline void add(int u,int v){    e[++cnt].v=v,e[cnt].nxt=head[u],head[u]=cnt;    e[++cnt].v=u,e[cnt].nxt=head[v],head[v]=cnt;}inline void update(int l,int r,int x,int &y,int num){    y=++sz,sum[y]=sum[x]+1;    if(l==r) return;    ls[y]=ls[x],rs[y]=rs[x];    int mid=(l+r)>>1;    if(num<=mid) update(l,mid,ls[x],ls[y],num);    else update(mid+1,r,rs[x],rs[y],num);} inline void dfs(int x,int ff){    update(1,tot,root[ff],root[x],v[x]);    f[x][0]=ff,dep[x]=dep[ff]+1;    for(int i=head[x];i;i=e[i].nxt){        int j=e[i].v;        if(j==ff) continue;        dfs(j,x);    }}inline void findf(){    for(int i=1;i<=16;i++) for(int j=1;j<=n;j++) if(dep[j]>=bin[i])f[j][i]=f[f[j][i-1]][i-1];}inline int LCA(int x,int y){    if(dep[x]<dep[y]) swap(x,y);    int set=dep[x]-dep[y];    for(int i=16;i>=0;i--) if(set>=bin[i]) {        x=f[x][i];        set-=bin[i];    }    if(x==y) return x;    for(int i=16;i>=0;i--) if(f[x][i]!=f[y][i]){        x=f[x][i],y=f[y][i];    }    return f[x][0];}inline int find(int x){    int l=1,r=tot;    while(l<=r){        int mid=(l+r)>>1;        if(hash[mid]<x)l=mid+1;        else if(hash[mid]==x)return mid;        else r=mid-1;    }}inline int query(int a,int b,int c,int d,int k){    int l=1,r=tot;     while(l<r){        int mid=(l+r)>>1;        int zjq=sum[ls[a]]+sum[ls[b]]-sum[ls[c]]-sum[ls[d]];        if(zjq>=k) r=mid,a=ls[a],b=ls[b],c=ls[c],d=ls[d];        else k-=zjq,l=mid+1,a=rs[a],b=rs[b],c=rs[c],d=rs[d];    }    return l;}int main(){#ifdef Devil_Gary    freopen("in.txt","r",stdin);#endif    for(int i=1;i<=16;i++) bin[i]=bin[i-1]*2;    n=read(),m=read();    for(int i=1;i<=n;i++) v[i]=read(),tmp[i]=v[i];    sort(tmp+1,tmp+n+1);    hash[++tot]=tmp[1];    for(int i=2;i<=n;i++)if(tmp[i]!=tmp[i-1])hash[++tot]=tmp[i];    for(int i=1;i<=n;i++)v[i]=find(v[i]);    for(int i=1;i<n;i++) add(read(),read());    dfs(1,0);    findf();    for(int i=1,x,y,z;i<=m;i++){        x=read()^lastans,y=read(),z=read();        int lca=LCA(x,y),flca=f[lca][0];        printf("%d",lastans=hash[query(root[x],root[y],root[lca],root[flca],z)]);        if(i!=m)printf("\n");    }    return 0;}

阅读全文

0 0