BZOJ2588: Spoj 10628. Count on a tree

题目链接

在树上建主席树+LCA，主席树中每一棵线段树维护的是当前点到根这一段区间。
我用的树链剖分求LCA，也可以用倍增。
（查询里带了七个参感觉很蠢。。）

【代码】

#include <cstdio>
#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <stack>
#define N 100005 
#define M 2001005
#define INF 1e9
#define mod 1000000007
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pa;

int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

int n,m,cnt,tot,ind,Cnt,Last_Ans;
int a[N],num[N],hash[N];
int ls[M],rs[M],root[N],size[M];
int sz[N],son[N],top[N],fa[N],pos[N],ftree[N],deep[N];
int b[N<<1],p[N],nextedge[N<<1];

void Add(int x,int y)
{
    Cnt++;
    b[Cnt]=y;
    nextedge[Cnt]=p[x];
    p[x]=Cnt;
}

void Anode(int x,int y){
    Add(x,y);Add(y,x);
}

void Input_Init()
{
    n=read(),m=read();
    for(int i=1;i<=n;i++) num[i]=a[i]=read();
    sort(num+1,num+1+n);
    for(int i=1;i<=n;i++) if(num[i]!=num[i-1]||i==1) hash[++tot]=num[i];
    for(int i=1;i<n;i++)
    {
        static int x,y;
        x=read(),y=read();
        Anode(x,y);
    }
}

void Dfs(int x)
{
    sz[x]=1;
    for(int i=p[x];i;i=nextedge[i])
    {
        int v=b[i];
        if(v==fa[x]) continue;
        fa[v]=x;deep[v]=deep[x]+1;
        Dfs(v);
        sz[x]+=sz[v];
        son[x]=sz[son[x]]>sz[v]?son[x]:v;
    }
}

void dfs(int x,int rt)
{
    pos[x]=++ind,ftree[pos[x]]=x,top[x]=rt;
    if(!son[x]) return;
    dfs(son[x],rt);
    for(int i=p[x];i;i=nextedge[i])
    {
        int v=b[i];
        if(v!=fa[x]&&v!=son[x]) dfs(v,v);
    }
}

int Find(int x)
{
    int l=1,r=tot,rtn;
    while(l<=r)
    {
        int mid=l+r>>1;
        if(hash[mid]>=x) rtn=mid,r=mid-1;
        else l=mid+1;
    }
    return rtn;
}

void Insert(int l,int r,int x,int &y,int z)
{
    y=++cnt;size[y]=size[x]+1;
    if(l==r) return;
    ls[y]=ls[x],rs[y]=rs[x];
    int mid=l+r>>1;
    if(z<=mid) Insert(l,mid,ls[x],ls[y],z);
    else Insert(mid+1,r,rs[x],rs[y],z);
}

void Build(int x)
{
    Insert(1,tot,root[fa[x]],root[x],Find(a[x]));
    for(int i=p[x];i;i=nextedge[i])
    {
        int v=b[i];
        if(v!=fa[x]) Build(v);
    }
}

void Pre_Work()
{
    Dfs(1);dfs(1,1);
    Build(1);
}

int Lca(int x,int y)
{
    int fx=top[x],fy=top[y];
    while(fx!=fy)
    {
        if(deep[fx]<deep[fy]) swap(x,y),swap(fx,fy);
        x=fa[fx],fx=top[x];
    }
    return deep[x]>deep[y]?y:x;
}

int Solve_Query(int l,int r,int A,int B,int C,int D,int k)
{
    if(l==r) return l;
    int mid=l+r>>1;
    int sum=size[ls[A]]+size[ls[B]]-size[ls[C]]-size[ls[D]];
    if(sum>=k) return Solve_Query(l,mid,ls[A],ls[B],ls[C],ls[D],k);
    return Solve_Query(mid+1,r,rs[A],rs[B],rs[C],rs[D],k-sum);
}

int Query(int x,int y,int k)
{
    int z=Lca(x,y),Anc=fa[z];
    return hash[Solve_Query(1,tot,root[x],root[y],root[z],root[Anc],k)];
}

void Solve()
{
    while(m--)
    {
        static int x,y,z;
        x=read(),y=read(),z=read();
        x^=Last_Ans;
        Last_Ans=Query(x,y,z);
        printf("%d",Last_Ans);
        if(m) printf("\n");
    }
}

int main()
{
    Input_Init();
    Pre_Work();
    Solve();
    return 0;
}