【bzoj2588】Count on a tree 主席树

这题给人开了个新思路。

原本构造一个序列的主席树，是这个位置用上个位置的信息来省空间，树上的主席树是继承父亲的信息来省空间。

此题若带修改怎么办？

若对某个点的权值做修改，则这个点的子树都会受影响，想到了dfs序上的树状数组加差分套权值线段树，感觉很劲啊。

update:

好吧，带修改的树上第K大：bzoj1146，确实有这么一种做法。

/**************************************************************
    Problem: 2588
    User: chad
    Language: C++
    Result: Accepted
    Time:6236 ms
    Memory:57688 kb
****************************************************************/
 
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<iomanip>
#include<map>
#include<queue>
using namespace std;
#define mem1(i,j) memset(i,j,sizeof(i))
#define mem2(i,j) memcpy(i,j,sizeof(i))
#define LL long long
#define up(i,j,n) for(LL i=(j);i<=(n);i++)
#define FILE "dealing"
#define poi vec
#define eps 1e-10
#define db double
#define mid ((l+r)>>1)
const int maxn=102000,inf=1000000000,mod=1000000007;
int read(){
    LL x=0,f=1,ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0',ch=getchar();}
    return f*x;
}
bool cmax(int& a,int b){return a<b?a=b,true:false;}
bool cmin(int& a,int b){return a>b?a=b,true:false;}
int n,m,L=1;
int fa[maxn][32],dep[maxn],v[maxn],t[maxn];
struct node{int y,next;}e[maxn<<1];int linkk[maxn<<1],len;
struct Node{int v,id;}a[maxn];
bool cmp(Node a,Node b){return a.v<b.v;}
bool cmp2(Node a,Node b){return a.id<b.id;}
void insert(int x,int y){e[++len].y=y;e[len].next=linkk[x];linkk[x]=len;}
int rt[2300000],cnt=0,sum[2300000],c[2300000][2];
int key=0;
void updata(int o){sum[o]=sum[c[o][1]]+sum[c[o][0]];}
void set(int& o,int x,int l,int r){
    o=++cnt;
    if(l==r){sum[o]=sum[x]+1;return;}
    if(key>mid)c[o][0]=c[x][0],set(c[o][1],c[x][1],mid+1,r);
    else c[o][1]=c[x][1],set(c[o][0],c[x][0],l,mid);
    updata(o);
}
void dfs(int x){
    key=v[x];set(rt[x],rt[fa[x][0]],1,L);
    for(int i=linkk[x];i;i=e[i].next){
        if(e[i].y==fa[x][0])continue;
        fa[e[i].y][0]=x;
        dep[e[i].y]=dep[x]+1;
        dfs(e[i].y);
    }
}
int lca(int x,int y){
    if(dep[x]>dep[y])swap(x,y);
    for(int i=28;i>=0;i--)if(dep[y]-dep[x]>=(1<<i))y=fa[y][i];
    if(x==y)return x;
    for(int i=28;i>=0;i--)if(fa[y][i]!=fa[x][i])x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int q[10],val,bol[10]={0,1,1,-1,-1};
int query(int l,int r){
    if(l==r)return t[l];
    val=0;up(i,1,4)val+=bol[i]*sum[c[q[i]][0]];
    //printf("%d %d %d %d\n",l,r,sum[q[1]]+sum[q[2]]-sum[q[3]]-sum[q[4]],0);
    if(key>=val){
        key-=val;
        up(i,1,4)q[i]=c[q[i]][1];
        return query(mid+1,r);}
    else {
        up(i,1,4)q[i]=c[q[i]][0];
        return query(l,mid);
    }
}
int getK(int x,int y,int k){
    key=k;
    q[1]=rt[x],q[2]=rt[y],q[3]=rt[k=lca(x,y)],q[4]=rt[fa[k][0]];
    return query(1,L);
}
int main(){
    //freopen(FILE".in","r",stdin);
    //freopen(FILE".out","w",stdout);
    n=read(),m=read();
    up(i,1,n)a[i].v=read(),a[i].id=i;
    up(i,1,n-1){
        int x=read(),y=read();
        insert(x,y);insert(y,x);
    }
    sort(a+1,a+n+1,cmp);
    int j=1;
    while(true){
        t[L]=a[j].v;//离散权值的实际权值
        v[a[j].id]=L;j++;//j节点的离散权值
        if(j>n)break;
        if(a[j].v!=a[j-1].v)L++;
    }
    sort(a+1,a+n+1,cmp2);
    dfs(1);
    for(int j=1;j<=30;j++)
        for(int i=1;i<=n;i++)
            fa[i][j]=fa[fa[i][j-1]][j-1];
    int last=0;
    while(m--){
        int u=read(),v=read(),k=read();
        u^=last;
        if(m)printf("%d\n",last=getK(u,v,k-1));
        else printf("%d",getK(u,v,k-1));
    }
    return 0;
}