bzoj2588 Count on a tree 可持久化线段树+lca

Description

---

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

N,M<=100000
强制在线

Solution

---

可持久化线段树不仅可以用在序列上，还可以在树上做。具体操作就是以每个点的父亲作为上一版本建新的树
考虑查询的两点为x,y，若lca(x,y)=x那么查询root[y]-root[fa[x]]，这点与序列上无异。推广一下就是任意两点x,y的查询都相当于查询root[x]-root[lca]+root[y]-root[lca]+root[lca]-root[fa[lca]]

几天不敲可持久化发现不太会了，一开始re是数组没开够

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define drp(i,st,ed) for (int i=st;i>=ed;--i)
#define fill(x,t) memset(x,t,sizeof(x))
const int N=200005;
const int E=400005;
struct treeNode{int l,r,c;}t[N*21];
struct edge{int x,y,next;}e[E];
int a[N],b[N],rank[N];
int fa[N][21],dep[N];
int ls[N],edCnt=0;
int root[N],cnt=0;
int lastans=0,size,n,m;
int read() {
    int x=0,v=1; char ch=getchar();
    for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
    for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
    return x*v;
}
void addEdge(int x,int y) {
    e[++edCnt]=(edge){x,y,ls[x]}; ls[x]=edCnt;
    e[++edCnt]=(edge){y,x,ls[y]}; ls[y]=edCnt;
}
int query(int x,int y,int lca,int lcaf,int l,int r,int k) {
    if (l==r) return l;
    int c=t[t[x].l].c+t[t[y].l].c-t[t[lca].l].c-t[t[lcaf].l].c,mid=(l+r)>>1;
    if (c>=k) return query(t[x].l,t[y].l,t[lca].l,t[lcaf].l,l,mid,k);
    else return query(t[x].r,t[y].r,t[lca].r,t[lcaf].r,mid+1,r,k-c);
}
void modify(int &now,int pre,int l,int r,int pos) {
    now=++cnt;
    t[now]=(treeNode){t[pre].l,t[pre].r,t[pre].c+1};
    if (l==r) return ;
    int mid=(l+r)>>1;
    if (pos<=mid) modify(t[now].l,t[pre].l,l,mid,pos);
    else modify(t[now].r,t[pre].r,mid+1,r,pos);
}
void buildTree(int &now,int l,int r) {
    now=++cnt;
    t[now]=(treeNode){0,0,0};
    if (l==r) return ;
    int mid=(l+r)>>1;
    buildTree(t[now].l,l,mid);
    buildTree(t[now].r,mid+1,r);
}
void dfs(int now) {
    modify(root[now],root[fa[now][0]],1,size,rank[now]);
    rep(i,1,20) fa[now][i]=fa[fa[now][i-1]][i-1];
    for (int i=ls[now];i;i=e[i].next) {
        if (e[i].y==fa[now][0]) continue;
        dep[e[i].y]=dep[now]+1;
        fa[e[i].y][0]=now;
        dfs(e[i].y);
    }
}
int get_lca(int x,int y) {
    if (dep[x]<dep[y]) std:: swap(x,y);
    drp(i,20,0) if (dep[fa[x][i]]>=dep[y]) x=fa[x][i];
    if (x==y) return x;
    drp(i,20,0) if (fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
void solve(int x,int y,int k) {
    int lca=get_lca(x,y);
    lastans=query(root[x],root[y],root[lca],root[fa[lca][0]],1,size,k);
    printf("%d\n", b[lastans]);
}
int main(void) {
    n=read(),m=read();
    rep(i,1,n) a[i]=b[i]=read();
    std:: sort(b+1,b+n+1);
    size=std:: unique(b+1,b+n+1)-b-1;
    rep(i,1,n) rank[i]=std:: lower_bound(b+1,b+size+1,a[i])-b;
    rep(i,2,n) {
        int x=read(),y=read();
        addEdge(x,y);
    }
    buildTree(root[0],1,size);
    dep[1]=1;
    dfs(1);
    while (m--) {
        int x=read(),y=read(),k=read();
        x^=b[lastans];
        solve(x,y,k);
    }
    return 0;
}