【bzoj 2588】Spoj 10628. Count on a tree

Description

给定一棵N个节点的树，每个点有一个权值，对于M个询问(u,v,k)，你需要回答u xor lastans和v这两个节点间第K小的点权。其中lastans是上一个询问的答案，初始为0，即第一个询问的u是明文。

Input

第一行两个整数N,M。

第二行有N个整数，其中第i个整数表示点i的权值。

后面N-1行每行两个整数(x,y)，表示点x到点y有一条边。

最后M行每行两个整数(u,v,k)，表示一组询问。

Output

M行，表示每个询问的答案。最后一个询问不输出换行符

Sample Input

8 5
105 2 9 3 8 5 7 7
1 2
1 3
1 4
3 5
3 6
3 7
4 8
2 5 1
0 5 2
10 5 3
11 5 4
110 8 2

Sample Output

2
8
9
105
7

HINT

N,M<=100000

暴力自重。。。

 

 

在树上建主席树，统计答案时为避免LCA被减两次，计算sum[x]+sum[y]-sum[lca(x,y)]-sum[fa[lca(x,y)]]。

 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 #define LL long long
 using namespace std;
 const int N=1e5+;
 int n,m,tot,x,y,rk,cnte,ind,cnt,temp,lastans;
 int v[N],tmp[N],first[N],id[N],num[N],root[N];
 int deep[N],fa[N][];
 struct node{int lc,rc,sum;}tr[N*];
 struct edge{int to,next;}e[N*];
 int read()
 {
     int x=,f=;char c=getchar();
     while(c<''||c>''){if(c=='-')f=-;c=getchar();}
     while(c>=''&&c<=''){x=x*+c-'';c=getchar();}
     return x*f;
 }
 void ins(int u,int v){e[++cnte]=(edge){v,first[u]};first[u]=cnte;}
 void insert(int u,int v){ins(u,v);ins(v,u);}
 void dfs(int x)
 {
     ind++;id[x]=ind;num[ind]=x;
     for(int i=;(<<i)<=deep[x];i++)fa[x][i]=fa[fa[x][i-]][i-];
     for(int i=first[x];i;i=e[i].next)
     {
         if(deep[e[i].to])continue;
         fa[e[i].to][]=x;deep[e[i].to]=deep[x]+;dfs(e[i].to);
     }
 }
 int lca(int ri,int rj)
 {
     if(deep[ri]<deep[rj])swap(ri,rj);
     int d=deep[ri]-deep[rj];
     for(int i=;(<<i)<=d;i++)if((<<i)&d)ri=fa[ri][i];
     if(ri==rj)return ri;
     for(int i=;i>=;i--)
         if((<<i)<=deep[ri]&&fa[ri][i]!=fa[rj][i])
             ri=fa[ri][i],rj=fa[rj][i];
     return fa[ri][];
 }
 void update(int &x,int last,int L,int R,int num)
 {
     x=++cnt;tr[x].sum=tr[last].sum+;
     if(L==R)return;
     tr[x].lc=tr[last].lc;tr[x].rc=tr[last].rc;
     int mid=(L+R)>>;
     if(num<=mid)update(tr[x].lc,tr[last].lc,L,mid,num);
     else update(tr[x].rc,tr[last].rc,mid+,R,num);
 }
 int query(int x,int y,int rk)
 {
     int a=x,b=y,c=lca(x,y),d=fa[c][];
     a=root[id[x]];b=root[id[b]];c=root[id[c]];d=root[id[d]];
     int L=,R=tot;
     while(L<R)
     {
         int mid=(L+R)>>;
         temp=tr[tr[a].lc].sum+tr[tr[b].lc].sum-tr[tr[c].lc].sum-tr[tr[d].lc].sum;
         if(temp>=rk)R=mid,a=tr[a].lc,b=tr[b].lc,c=tr[c].lc,d=tr[d].lc;
         else rk-=temp,L=mid+,a=tr[a].rc,b=tr[b].rc,c=tr[c].rc,d=tr[d].rc;
     }
     return tmp[L];
 }
 int main()
 {
     n=read();m=read();
     for(int i=;i<=n;i++)v[i]=tmp[i]=read();
     sort(tmp+,tmp+n+);tot=unique(tmp+,tmp+n+)-tmp-;
     for(int i=;i<=n;i++)v[i]=lower_bound(tmp+,tmp+tot+,v[i])-tmp;
     for(int i=;i<n;i++)x=read(),y=read(),insert(x,y);
     deep[]=;dfs();
     for(int i=;i<=n;i++)
     {
         temp=num[i];
         update(root[i],root[id[fa[temp][]]],,tot,v[temp]);
     }
     for(int i=;i<=m;i++)
     {
         x=read();y=read();rk=read();x^=lastans;
         lastans=query(x,y,rk);printf("%d",lastans);
         if(i!=m)printf("\n");
     }
     return ;
 }