[bzoj2588][主席树]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

暴力自重。。。

题解

这个强制在线。。然而又有区间k大值，于是乎就上主席树吧
对于每个孩子，先建主席树，然后和他的父亲合起来。就相当于每个点的主席树存储了他到根这条链的信息
至于查询，找到lca和lca的父亲。tr[tr[x].lc].c+tr[tr[y].lc].c-tr[tr[lca]lc].c-tr[tr[fa[lca]].lc].c>k的话就说明左孩子可以查，不然就进右孩子
这题有PE风险的

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
struct edge{int x,y,next;}a[211000];int last[111000],len;
struct LSnode{int y,p;}w[111000];
struct node
{
    int lc,rc,c;
}tr[2110000];int tot;
bool cmp(LSnode n1,LSnode n2){return n1.y<n2.y;}
int n,m,lastans;
int Rank[111000],tmp[111000];//第i个排第几 
void ins(int x,int y)
{
    len++;
    a[len].x=x;a[len].y=y;
    a[len].next=last[x];last[x]=len;
}
int fa[111000][25],dep[111000],bin[25],rt[111000];
void add(int &now,int l,int r,int p)
{
    if(now==0)now=++tot;
    tr[now].c++;
    if(l==r)return ;
    int mid=(l+r)/2;
    if(p<=mid)add(tr[now].lc,l,mid,p);
    else add(tr[now].rc,mid+1,r,p);
}
void merge(int &x,int y)
{
    if(x==0){x=y;return ;}
    if(y==0)return ;
    tr[x].c+=tr[y].c;
    merge(tr[x].lc,tr[y].lc);
    merge(tr[x].rc,tr[y].rc);
}
void pre_tree_node(int x)
{
    for(int i=1;bin[i]<=dep[x];i++)fa[x][i]=fa[fa[x][i-1]][i-1];
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(y!=fa[x][0])
        {
            add(rt[y],1,n,Rank[y]);
            merge(rt[y],rt[x]);
            fa[y][0]=x;
            dep[y]=dep[x]+1;
            pre_tree_node(y);
        }
    }
}
int lca(int x,int y)
{
    if(dep[x]<dep[y])swap(x,y);
    for(int i=20;i>=0;i--)if(dep[fa[x][i]]>=dep[y] && bin[i]<=dep[x])x=fa[x][i];
    if(x==y)return x;
    for(int i=20;i>=0;i--)if(bin[i]<=dep[x] && fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int get_K(int u,int v,int p,int q,int l,int r,int k)
{
    if(l==r)return w[l].y;
    int mid=(l+r)/2;
    int c=tr[tr[u].lc].c+tr[tr[v].lc].c-tr[tr[p].lc].c-tr[tr[q].lc].c;
    if(k<=c)return get_K(tr[u].lc,tr[v].lc,tr[p].lc,tr[q].lc,l,mid,k);
    else return get_K(tr[u].rc,tr[v].rc,tr[p].rc,tr[q].rc,mid+1,r,k-c);
}
int main()
{
    bin[0]=1;for(int i=1;i<=20;i++)bin[i]=bin[i-1]*2;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){scanf("%d",&tmp[i]);w[i].y=tmp[i];w[i].p=i;}
    sort(w+1,w+1+n,cmp);
    for(int i=1;i<=n;i++)Rank[w[i].p]=i;
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        ins(x,y);ins(y,x);
    }
    lastans=0;
    fa[1][0]=0;dep[1]=1;
    add(rt[1],1,n,Rank[1]);
    pre_tree_node(1);
    while(m--)
    {
        int x,y,k;
        scanf("%d%d%d",&x,&y,&k);
        x^=lastans;
        int u=lca(x,y),v=fa[u][0];
        lastans=get_K(rt[x],rt[y],rt[u],rt[v],1,n,k);
        printf("%d",lastans); 
        if(m)printf("\n");
    }
    return 0;
}