BZOJ 2588 主席树

来源:互联网 发布:何念php 编辑:程序博客网 时间:2024/05/29 05:01

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

//By SiriusRen#include <cstdio>#include <cstring>#include <algorithm>using namespace std;#define N 200050#define M 5000000typedef 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');    }}

这里写图片描述

0 0
原创粉丝点击