bzoj 4771: 七彩树

题意：

给定一棵n个点的有根树，编号依次为1到n，其中1号点是根节点。每个节点都被染上了某一种颜色，其中第i个节点的颜色为c[i]。如果c[i]=c[j]，那么我们认为点i和点j拥有相同的颜色。定义depth[i]为i节点与根节点的距离，为了方便起见，你可以认为树上相邻的两个点之间的距离为1。站在这棵色彩斑斓的树前面，你将面临m个问题。
每个问题包含两个整数x和d，表示询问x子树里且depth不超过depth[x]+d的所有点中出现了多少种本质不同的颜色。请写一个程序，快速回答这些询问。

题解:

好题，一眼$log^2$树套树，感觉没有前途懒得写了。
对dfs序的理解停留在子树是连续区间阶段
考虑没有深度限制怎么做不是区间不同的数的sb做法
对于相同颜色的点按dfs序排个序，现将每个位置+1，然后将相邻点的lca-1，因为dfs序的性质，相邻lca的一定是最深的大概YY一下。这样就转换为区间求和问题。
那么有了深度限制，因为没有修改，所以按深度排好序，一个个插入主席树即可。
读优+输优水过
code：

#include<set>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
struct trnode{
    int lc,rc,c;
}tr[6000010];int tot=0,root[100010];
struct node{
    int max,min;
    set<int> s;
    node() {max=min=0;}
}s[100010];
struct Node{
    int y,next;
}a[200010];int len=0,last[100010];
int ys[100010],la[100010],z=0,num[100010];
int Fa[18][100010],dep[100010],t[100010];
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,m,c[100010];
void ins(int x,int y)
{
    a[++len].y=y;
    a[len].next=last[x];last[x]=len;
}
void dfs(int x,int fa)
{
    ys[x]=la[x]=++z;t[z]=x;
    Fa[0][x]=fa;dep[x]=dep[fa]+1;
    for(int i=1;(1<<i)<=dep[x];i++)
        Fa[i][x]=Fa[i-1][Fa[i-1][x]];
    for(int i=last[x];i;i=a[i].next)
    {
        int y=a[i].y;
        if(y==fa) continue;
        dfs(y,x);la[x]=la[y];
    }
}
int findlca(int x,int y)
{
    if(dep[x]<dep[y]) swap(x,y);
    for(int i=16;i>=0;i--)
        if((1<<i)<=dep[x]-dep[y]) x=Fa[i][x];
    if(x==y) return x;
    for(int i=16;i>=0;i--)
        if((1<<i)<=dep[x]&&Fa[i][x]!=Fa[i][y])
            x=Fa[i][x],y=Fa[i][y];
    return Fa[0][x]; 
}
void update(int &x,int fr,int l,int r,int k,int c)
{
    x=++tot;tr[x]=tr[fr];
    if(l==r) {tr[x].c+=c;return;}
    int mid=(l+r)/2;
    if(k<=mid) update(tr[x].lc,tr[fr].lc,l,mid,k,c);
    else update(tr[x].rc,tr[fr].rc,mid+1,r,k,c);
    tr[x].c=tr[tr[x].lc].c+tr[tr[x].rc].c;
}
int findans(int x,int l,int r,int fl,int fr)
{
    if(l==fl&&r==fr) return tr[x].c;
    int mid=(l+r)/2;
    if(fr<=mid) return findans(tr[x].lc,l,mid,fl,fr);
    if(fl>mid) return findans(tr[x].rc,mid+1,r,fl,fr);
    return findans(tr[x].lc,l,mid,fl,mid)+findans(tr[x].rc,mid+1,r,mid+1,fr);
}
bool cmp(int a,int b) {return dep[a]<dep[b];}
void add(int x,int d)
{
    if(s[c[x]].min==0&&s[c[x]].max==0)
    {
        s[c[x]].min=s[c[x]].max=ys[x];
        update(root[d],root[d],1,n,ys[x],1);
        s[c[x]].s.insert(ys[x]);
    }
    else if(ys[x]<s[c[x]].min)
    {
        s[c[x]].min=ys[x];
        update(root[d],root[d],1,n,ys[x],1);
        update(root[d],root[d],1,n,ys[findlca(t[*s[c[x]].s.begin()],x)],-1);
        s[c[x]].s.insert(ys[x]);
    }
    else if(ys[x]>s[c[x]].max)
    {
        s[c[x]].max=ys[x];
        update(root[d],root[d],1,n,ys[x],1);
        update(root[d],root[d],1,n,ys[findlca(t[*(--s[c[x]].s.end())],x)],-1);
        s[c[x]].s.insert(ys[x]);
    }
    else
    {
        int l=*(--s[c[x]].s.lower_bound(ys[x])),r=*(s[c[x]].s.lower_bound(ys[x]));
        update(root[d],root[d],1,n,ys[x],1);
        update(root[d],root[d],1,n,ys[findlca(t[l],t[r])],1);
        update(root[d],root[d],1,n,ys[findlca(t[l],x)],-1);
        update(root[d],root[d],1,n,ys[findlca(t[r],x)],-1);
        s[c[x]].s.insert(ys[x]);
    }
}
inline void write(int x)
{
    if (!x) return (void)puts("0");
    if (x < 0) putchar('-'), x = -x;
    static short s[12], t;
    while (x) s[++t] = x % 10, x /= 10;
    while (t) putchar('0' + s[t--]);
    putchar('\n');
}
int main()
{
    int T;T=read();
    while(T--)
    {
        n=read();m=read();tot=z=len=tot=0;memset(last,0,sizeof(last));
        for(int i=1;i<=n;i++) c[i]=read(),num[i]=i;
        for(int i=2;i<=n;i++)
        {
            int fa=read();
            ins(fa,i);ins(i,fa);
        }
        for(int i=1;i<=n;i++) s[i].min=s[i].max=0,s[i].s.clear();
        dep[0]=-1;dfs(1,0);
        sort(num+1,num+n+1,cmp);
        int p=1,D=0;
        while(p<=n)
        {
            D++;root[D]=root[D-1];
            do {add(num[p],D);p++;}
            while(p<=n&&dep[num[p]]==dep[num[p-1]]);
        }
        int ans=0;
        for(int i=1;i<=m;i++)
        {
            int x=read()^ans,d=read()^ans;
            ans=findans(root[min(D,dep[x]+d+1)],1,n,ys[x],la[x]);
            write(ans);
        }
    }
}