bzoj4771 七彩树

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

#include<cstdio>#include<set>#include<vector>#include<algorithm>using namespace std;#define LL long longint 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();}