bzoj2588: Spoj 10628. Count on a tree（主席樹+LCA）

題目傳送門
。

解法：
樹上主席樹。
我在想要不要離散化。
想看看資料範圍。
翻了翻discuss發現很神奇啊。
什麼MLE，什麼開longlong？？
然後我就很乖的開了long long。

第i棵主席樹維護根到i的資訊。
如何求x到y路徑的資訊呢。
p表示lca。
那麼用x到根+y到根-p到根-p的父親到根就得到x到y的資訊。
然後作第K大就行了呀。

記得離散化。

程式碼實現：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
struct node {int lc,rc,c;}t[5100000];int cnt,rt[110000];
void build(int &u,int l,int r,int p) {
    if(u==0)u=++cnt;t[u].c++;
    if(l==r)return ;int mid=(l+r)/2;
    if(p<=mid)build(t[u].lc,l,mid,p);
    else build(t[u].rc,mid+1,r,p);
}
void Merge(int &u1,int u2) {
    if(u1==0) {u1=u2;return ;}if(u2==0)return ;
    t[u1].c+=t[u2].c;
    Merge(t[u1].lc,t[u2].lc);
    Merge(t[u1].rc,t[u2].rc);
}
struct trnode {int x,y,next;}a[210000];int len,last[110000];
void ins(int x,int y) {len++;a[len].x=x;a[len].y=y;a[len].next=last[x];last[x]=len;}
int mx[21][110000],dep[110000];
void dfs(int x) {
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(y!=mx[0][x]) {
            mx[0][y]=x;dep[y]=dep[x]+1;
            dfs(y);
        }
    }
}
int bin[21],n;
void work() {
    bin[0]=1;for(int i=1;i<=20;i++)bin[i]=bin[i-1]*2;
    for(int j=1;j<=20;j++)for(int i=1;i<=n;i++)if(dep[i]>=bin[j])
        mx[j][i]=mx[j-1][mx[j-1][i]];
}
void merge(int x) {
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(y!=mx[0][x]){Merge(rt[y],rt[x]);merge(y);}
    }
}ll S[110000];
struct edge {ll x;int id;}s[110000];
bool cmp(edge n1,edge n2) {return n1.x<n2.x;}
int lca(int x,int y) {
    if(dep[x]>dep[y])swap(x,y);
    for(int i=20;i>=0;i--)if(dep[y]-dep[x]>=bin[i])y=mx[i][y];
    if(x==y)return x;
    for(int i=20;i>=0;i--)if(mx[i][x]!=mx[i][y]&&dep[x]>=bin[i]) {
        x=mx[i][x];y=mx[i][y];
    }return mx[0][x];
}
ll find(int u1,int u2,int u3,int u4,int l,int r,int k) {
    if(l==r)return S[l];
    int mid=(l+r)/2;
    int c=t[t[u1].lc].c+t[t[u2].lc].c-t[t[u3].lc].c-t[t[u4].lc].c;
    if(c>=k) return find(t[u1].lc,t[u2].lc,t[u3].lc,t[u4].lc,l,mid,k);
    else return find(t[u1].rc,t[u2].rc,t[u3].rc,t[u4].rc,mid+1,r,k-c);
}
ll ans[110000];
int main() {
    int m;scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) {scanf("%lld",&s[i].x);s[i].id=i;}
    sort(s+1,s+1+n,cmp);int tot=0;
    for(int i=1;i<=n;i++) {
        if(s[i].x!=s[i-1].x)tot++;S[tot]=s[i].x;
        build(rt[s[i].id],1,n,tot);
    }len=0;memset(last,0,sizeof(last));
    for(int i=1;i<n;i++) {
        int x,y;scanf("%d%d",&x,&y);
        ins(x,y);ins(y,x);
    }mx[0][1]=0;dep[1]=0;dfs(1);work();merge(1);
    ll lastans=0ll;
    for(int i=1;i<=m;i++) {
        int x,y,k;scanf("%d%d%d",&x,&y,&k);x^=lastans;
        int Lca=lca(x,y);
        lastans=find(rt[x],rt[y],rt[Lca],rt[mx[0][Lca]],1,n,k);
        ans[i]=lastans;
    }
    for(int i=1;i<m;i++)printf("%lld\n",ans[i]);
    printf("%lld",ans[m]);
    return 0;
}