JZOJ4444

题目大意

询问树上两点的路径上小于/等于/大于所给定值k得点的数量（每次询问k不一定相同）
强制在线

主席树

以计算小于k为例：
u到v的路径上小于k的点的数量=>u到根节点小于k的点的数量+v到根节点小于k的点的数量-2*lca(u,v)到根节点小于k的点的数量+lca(u,v)的权值是否小于k
然后裸的主席树

#include<cstdio>
#include<algorithm>
#include<cmath>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
const int maxn=300000+10,maxm=200000+10;
int root[maxn],left[maxn*20],right[maxn*20],sum[maxn*20],a[maxn];
int h[maxn],go[maxn*2],next[maxn*2];
int d[maxn],f[maxn][20];
bool czy;
int i,j,k,l,t,n,m,q,tot,now,ans,u,v,w;
void add(int x,int y){
    go[++tot]=y;
    next[tot]=h[x];
    h[x]=tot;
}
void insert(int &x,int y,int l,int r,int a){
    x=++tot;
    left[x]=left[y];
    right[x]=right[y];
    sum[x]=sum[y]+1;
    if (l==r) return;
    int mid=(l+r)/2;
    if (a<=mid) insert(left[x],left[y],l,mid,a);else insert(right[x],right[y],mid+1,r,a);
}
void dfs(int x,int y){
    d[x]=d[y]+1;
    f[x][0]=y;
    insert(root[x],root[y],1,m,a[x]);
    int t=h[x];
    while (t){
        if (go[t]!=y) dfs(go[t],x);
        t=next[t];
    }
}
int lca(int x,int y){
    if (d[x]<d[y]) swap(x,y);
    int j;
    if (d[x]!=d[y]){
        j=floor(log(d[x])/log(2));
        while (j>=0){
            if (d[f[x][j]]>d[y]) x=f[x][j];
            j--;
        }
        x=f[x][0];
    }
    if (x==y) return x;
    j=floor(log(d[x])/log(2));
    while (j>=0){
        if (f[x][j]!=f[y][j]){
            x=f[x][j];
            y=f[y][j];
        }
        j--;
    }
    return f[x][0];
}
int query(int x,int l,int r,int a,int b){
    if (a>b||!x) return 0;
    if (l==a&&r==b) return sum[x];
    int mid=(l+r)/2;
    if (b<=mid) return query(left[x],l,mid,a,b);
    else if (a>mid) return query(right[x],mid+1,r,a,b);
    else return query(left[x],l,mid,a,mid)+query(right[x],mid+1,r,mid+1,b);
}
int main(){
    czy=1;
    scanf("%d%d%d",&n,&m,&q);
    fo(i,1,n) scanf("%d",&a[i]);
    fo(i,1,n-1){
        scanf("%d%d",&j,&k);
        add(j,k);add(k,j);
    }
    tot=0;
    dfs(1,0);
    fo(j,1,floor(log(n)/log(2)))
        fo(i,1,n)
            f[i][j]=f[f[i][j-1]][j-1];
    while (q--){
        scanf("%d%d%d",&u,&v,&k);
        if (czy) u^=now,v^=now,k^=now;
        now=0;
        w=lca(u,v);
        ans=query(root[u],1,m,1,k-1)+query(root[v],1,m,1,k-1)-2*query(root[w],1,m,1,k-1);
        if (a[w]<k) ans++;
        printf("%d ",ans);
        now^=ans;
        ans=query(root[u],1,m,k,k)+query(root[v],1,m,k,k)-2*query(root[w],1,m,k,k);
        if (a[w]==k) ans++;
        printf("%d ",ans);
        now^=ans;
        ans=query(root[u],1,m,k+1,m)+query(root[v],1,m,k+1,m)-2*query(root[w],1,m,k+1,m);
        if (a[w]>k) ans++;
        printf("%d\n",ans);
        now^=ans;
    }
}