BZOJ 2588 主席树

思路:
主席树
做完BZOJ 3123 觉得这是道水啊……
然后狂RE 狂MLE
要来数据
忘把deep[1]设成1了………………………….
啊wocccccccccccccccc

//By SiriusRen
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 200050
#define M 5000000
typedef long long ll;
int n,m,xx,yy,zz,a[N],inf,cnt,fa[N/2][21],tree[M],lson[M],rson[M];
int first[N],next[N],v[N],root[N],tot,ans,jy,deep[N];
void add(int x,int y){v[tot]=y,next[tot]=first[x],first[x]=tot++;}
void insert(int x,int &y,int l,int r,int wei){
    if(!y)y=++cnt;tree[y]=tree[x]+1;if(l==r)return;
    int mid=((ll)l+r)/2;
    if(mid<wei)lson[y]=lson[x],insert(rson[x],rson[y],mid+1,r,wei);
    else rson[y]=rson[x],insert(lson[x],lson[y],l,mid,wei);
}
void dfs(int x){
    for(int i=1;i<=20;i++)fa[x][i]=fa[fa[x][i-1]][i-1];
    insert(root[fa[x][0]],root[x],1,inf,a[x]);
    for(int i=first[x];~i;i=next[i])if(v[i]!=fa[x][0]){
        deep[v[i]]=deep[x]+1;
        fa[v[i]][0]=x,dfs(v[i]);
    }
}
int lca(int x,int y){
    if(deep[x]<deep[y])swap(x,y);
    for(int i=20;~i;i--)if(deep[fa[x][i]]>=deep[y])x=fa[x][i];
    if(x==y)return x;
    for(int i=20;~i;i--)if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int query(int x,int y,int lca,int flca,int l,int r,int wei){
    if(l==r)return l;
    int mid=((ll)l+r)/2,temp=tree[lson[x]]+tree[lson[y]]-tree[lson[lca]]-tree[lson[flca]];
    if(temp<wei)return query(rson[x],rson[y],rson[lca],rson[flca],mid+1,r,wei-temp);
    else return query(lson[x],lson[y],lson[lca],lson[flca],l,mid,wei);
}
signed main(){
    memset(first,-1,sizeof(first)),scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]),inf=max(inf,a[i]);
    for(int i=1;i<n;i++)scanf("%d%d",&xx,&yy),add(xx,yy),add(yy,xx);
    deep[1]=1,dfs(1);
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&xx,&yy,&zz),xx^=ans;jy=lca(xx,yy);
        printf("%d",ans=query(root[xx],root[yy],root[jy],root[fa[jy][0]],1,inf,zz));
        if(i!=m)putchar('\n');
    }
}

这里写图片描述

posted @ 2017-01-12 17:44  SiriusRen  阅读(135)  评论(0编辑  收藏  举报