[BZOJ2588][SPOJ10628]Count on a tree

题目大意：
　　给定一棵$n(nleq100000)$个结点的带点权的树，有$m(mleq100000)$组询问，每次询问$u$到$v$的路径上第$k$小的权值。强制在线。

思路：
　　每个结点根据父结点建立可持久化权值线段树，查询时就查询$u+v-lca(u,v)-par[lca(u,v)]$的树即可。

#include<list>
#include<cstdio>
#include<cctype>
#include<algorithm>
inline int getint() {
    register char ch;
    while(!isdigit(ch=getchar()));
    register int x=ch^'0';
    while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
    return x;
}
const int N=100001,M=100000,logN=17;
int w[N],tmp[N],dep[N],anc[N][logN];
std::list<int> e[N];
inline void add_edge(const int &u,const int &v) {
    e[u].push_back(v);
    e[v].push_back(u);
}
class FotileTree {
    private:
        struct Node {
            int val;
            Node *left,*right;
            Node() {}
            Node(const Node *const &p) {
                *this=*p;
            }
        };
    public:
        Node *root[N];
        FotileTree() {
            root[0]=new Node;
            root[0]->left=root[0]->right=root[0];
        }
        Node *modify(const Node *const &last,const int &b,const int &e,const int &x) {
            Node *const p=new Node(last);
            p->val++;
            if(b==e) return p;
            const int mid=(b+e)>>1;
            if(x<=mid) p->left=modify(last->left,b,mid,x);
            if(x>mid) p->right=modify(last->right,mid+1,e,x);
            return p;
        }
        int query(const Node *const &p1,const Node *const &p2,const Node *const &p3,const Node *const &p4,const int &b,const int &e,const int &k) {
            if(b==e) return b;
            const int mid=(b+e)>>1;
            if(k<=p1->left->val+p2->left->val-p3->left->val-p4->left->val) return query(p1->left,p2->left,p3->left,p4->left,b,mid,k);
            return query(p1->right,p2->right,p3->right,p4->right,mid+1,e,k-(p1->left->val+p2->left->val-p3->left->val-p4->left->val));
        }
};
FotileTree t;
inline int log2(const float &x) {
    return ((unsigned&)x>>23&255)-127;
}
void dfs(const int &x,const int &par) {
    t.root[x]=t.modify(t.root[par],1,tmp[0],w[x]);
    anc[x][0]=par;
    dep[x]=dep[par]+1;
    for(register int i=1;i<=log2(dep[x]);i++) {
        anc[x][i]=anc[anc[x][i-1]][i-1];
    }
    for(std::list<int>::iterator i=e[x].begin();i!=e[x].end();i++) {
        const int &y=*i;
        if(y==par) continue;
        dfs(y,x);
    }
}
inline int get_lca(int u,int v) {
    if(dep[u]<dep[v]) std::swap(u,v);
    while(dep[u]!=dep[v]) {
        u=anc[u][log2(dep[u]-dep[v])];
    }
    if(u==v) return u;
    for(register int i=log2(dep[u]);~i;i--) {
        if(anc[u][i]!=anc[v][i]) {
            u=anc[u][i];
            v=anc[v][i];
        }
    }
    return anc[u][0];
}
int main() {
    const int n=getint(),m=getint();
    for(register int i=1;i<=n;i++) {
        w[i]=tmp[i]=getint();
    }
    std::sort(&tmp[1],&tmp[n]+1);
    tmp[0]=std::unique(&tmp[1],&tmp[n]+1)-&tmp[1];
    for(register int i=1;i<=n;i++) {
        w[i]=std::lower_bound(&tmp[1],&tmp[tmp[0]]+1,w[i])-&tmp[0];
    }
    for(register int i=1;i<n;i++) {
        add_edge(getint(),getint());
    }
    dfs(1,0);
    for(register int i=0,ans=0;i<m;i++) {
        const int u=getint()^ans,v=getint(),k=getint(),lca=get_lca(u,v);
        if(i) putchar('
');
        printf("%d",ans=tmp[t.query(t.root[u],t.root[v],t.root[lca],t.root[anc[lca][0]],1,tmp[0],k)]);
    }
    return 0;
}