bzoj4771 七彩树

如果每次只是询问子树的话，可以求出dfs序，对于每种颜色在出现处$+1$，在相邻两点的lca处$-1$，直接子树求和就可以了。
但是这时加入了深度限制，考虑对深度进行可持久化。建立以dfs序为权值的主席树，从上到下对每一层加入颜色进行修改。查询前驱后继可以用set维护。

#include<cstdio>
#include<set>
#include<vector>
#include<algorithm>
using namespace std;
#define LL long long
int rd()
{
    int x=0;
    char c=getchar();
    while (c<'0'||c>'9') c=getchar();
    while (c>='0'&&c<='9')
    {
        x=x*10+c-'0';
        c=getchar();
    }
    return x;
}
const int maxn=400010,maxt=10000010;
int c[maxn],dep[maxn],root[maxn],fa[maxn],L[maxn],R[maxn],
pos[maxn],mn[maxn][25],log[maxn],
lson[maxt],rson[maxt],sum[maxt],
n,m,q,tot,clo,now;
vector<int> son[maxn],a[maxn];
int cmpd(int u,int v)
{
    return dep[u]<dep[v];
}
struct cmpp
{
    bool operator () (const int u,const int v)
    {
        return pos[u]<pos[v];
    }
};
set<int,cmpp> s[maxn];
int lca(int u,int v)
{
    int x=pos[u],y=pos[v],k;
    if (x>y) swap(x,y);
    k=log[y-x+1];
    return min(mn[x][k],mn[y-(1<<k)+1][k],cmpd);
}
void dfs(int u)
{
    dep[u]=dep[fa[u]]+1;
    L[u]=++clo;
    mn[pos[u]=++now][0]=u;
    a[dep[u]].push_back(u);
    m=max(m,dep[u]);
    for (vector<int>::iterator it=son[u].begin();it!=son[u].end();++it)
    {
        dfs(*it);
        mn[++now][0]=u;
    }
    R[u]=clo;
}
int modi(int u,int L,int R,int p,int x)
{
    if (!u)
    {
        u=++tot;
        lson[u]=rson[u]=sum[u]=0;
    }
    sum[u]+=x;
    if (L==R) return u;
    int mid=L+R>>1;
    if (p<=mid) lson[u]=modi(lson[u],L,mid,p,x);
    else rson[u]=modi(rson[u],mid+1,R,p,x);
    return u;
}
int merge(int u,int v,int L,int R)
{
    if ((LL)u*v==0) return u+v;
    sum[u]+=sum[v];
    if (L==R) return u;
    int mid=L+R>>1;
    lson[u]=merge(lson[u],lson[v],L,mid);
    rson[u]=merge(rson[u],rson[v],mid+1,R);
    return u;
}
int qry(int u,int L,int R,int l,int r)
{
    if (!u) return 0;
    if (l<=L&&R<=r) return sum[u];
    int mid=L+R>>1,ret=0;
    if (l<=mid) ret+=qry(lson[u],L,mid,l,r);
    if (r>mid) ret+=qry(rson[u],mid+1,R,l,r);
    return ret;
}
void solve()
{
    int u,x,y,d,last=0;
    set<int>::iterator i1,i2;
    n=rd();
    q=rd();
    for (int i=1;i<=n;i++) son[i].clear(),a[i].clear(),s[i].clear(),root[i]=0;
    m=clo=tot=now=0;
    for (int i=1;i<=n;i++) c[i]=rd();
    for (int i=2;i<=n;i++)
    {
        fa[i]=rd();
        son[fa[i]].push_back(i);
    }
    dfs(1);
    for (int k=1;(1<<k)<=now;k++) log[1<<k]=k;
    for (int i=3;i<=now;i++)
        if (!log[i]) log[i]=log[i-1];
    for (int k=1;k<=log[now];k++)
        for (int i=1;i+(1<<k)-1<=now;i++)
            mn[i][k]=min(mn[i][k-1],mn[i+(1<<k-1)][k-1],cmpd);
    for (int i=1;i<=m;i++)
    {
        for (vector<int>::iterator it=a[i].begin();it!=a[i].end();++it)
        {
            u=*it;
            i1=s[c[u]].lower_bound(u);
            x=i1==s[c[u]].begin()?-1:*(--i1);
            i1=s[c[u]].lower_bound(u);
            y=i1==s[c[u]].end()?-1:*i1;
            if (x>0) root[i]=modi(root[i],1,n,L[lca(x,u)],-1);
            if (y>0) root[i]=modi(root[i],1,n,L[lca(y,u)],-1);
            if (x>0&&y>0) root[i]=modi(root[i],1,n,L[lca(x,y)],1);
            root[i]=modi(root[i],1,n,L[u],1);
            s[c[u]].insert(u);
        }
        root[i]=merge(root[i],root[i-1],1,n);
    }
    /*for (int i=1;i<=tot;i++)
        if (lson[i]>tot||rson[i]>tot) printf("hhh:%d:%d,%d\n",i,lson[i],rson[i]);*/
    while (q--)
    {
        u=rd()^last;
        d=rd()^last;
        /*u=rd();
        d=rd();*/
        d=min(m,dep[u]+d);
        //qry(root[d],1,n,L[u],R[u]);
        printf("%d\n",last=qry(root[d],1,n,L[u],R[u]));
    }
}
int main()
{
    //freopen("in","r",stdin);
    int T;
    T=rd();
    while (T--) solve();
}