[BZOJ2588][Spoj10628]Count on a tree（树上主席树）

题目：

我是超链接

题解：

给每个点建一棵主席树，维护到根节点的所有点，这样答案就是$sum(a)+sum(b)-sum(lca(a,b))-sum(fa(lca(a,b)))$

代码：

#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;
const int mc=18;
const int N=100005;
int tot,nxt[N*2],point[N],v[N*2],h[N],f[N][mc],dfn[N],num,mi[mc];
int b[N],root[N],sz,a[N],s,pos[N];
struct hh{int l,r,w;}t[N*mc*5];
void addline(int x,int y)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
    ++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x;
}
void dfs(int x,int fa)
{
    h[x]=h[fa]+1;dfn[++num]=x;
    for (int i=1;i<mc;i++)
      if (mi[i]>h[x]) break;
      else f[x][i]=f[f[x][i-1]][i-1];
    for (int i=point[x];i;i=nxt[i])
      if (v[i]!=fa)
      {
        f[v[i]][0]=x;
        dfs(v[i],x);
      }
}
int lca(int x,int y)
{
    if (h[x]<h[y]) swap(x,y);
    int k=h[x]-h[y];
    for (int i=0;i<mc;i++)
      if (k&(1<<i)) x=f[x][i];
    if (x==y) return x;
    for (int i=mc-1;i>=0;i--)
      if (f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    return f[x][0];
}
void insert(int &now,int l,int r,int x)
{
    t[++sz]=t[now]; now=sz;
    t[now].w++;
    if (l==r) return;
    int mid=(l+r)>>1;
    if (x<=mid) insert(t[now].l,l,mid,x);
    else insert(t[now].r,mid+1,r,x);
}
int qurry(int l,int r,int x,int y,int fa,int of,int k)
{
    if (l==r) return b[l];
    int mid=(l+r)>>1;
    if (t[t[x].l].w+t[t[y].l].w-t[t[fa].l].w-t[t[of].l].w>=k)
      return qurry(l,mid,t[x].l,t[y].l,t[fa].l,t[of].l,k);
    else 
      return qurry(mid+1,r,t[x].r,t[y].r,t[fa].r,t[of].r,k-(t[t[x].l].w+t[t[y].l].w-t[t[fa].l].w-t[t[of].l].w));
}
int main()
{
    int m,x,y,k,n,ans=0;mi[0]=1;
    for (int i=0;i<mc;i++) mi[i]=mi[i-1]*2;
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++) scanf("%d",&a[i]),b[++s]=a[i];
    sort(b+1,b+s+1);
    s=unique(b+1,b+s+1)-b-1;
    for (int i=1;i<=n;i++) pos[i]=lower_bound(b+1,b+s+1,a[i])-b;
    for (int i=1;i<n;i++) 
    {
        scanf("%d%d",&x,&y);
        addline(x,y);
    }
    dfs(1,0);
    for (int i=1;i<=num;i++)
    {
        root[dfn[i]]=root[f[dfn[i]][0]];
        insert(root[dfn[i]],1,s,pos[dfn[i]]);
    }
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&x,&y,&k);
        x^=ans;
        ans=qurry(1,s,root[x],root[y],root[lca(x,y)],root[f[lca(x,y)][0]],k);
        printf("%d",ans);
        if (i!=m) printf("\n");
    }
}