[BZOJ]2017省队十连测推广赛1 T2.七彩树

题目大意：给你一棵n个点的树，每个点有颜色，m次询问，每次询问一个点x的子树内深度不超过depth[x]+d的节点的颜色数量，强制在线。（n,m<=100000，多组数据，保证n,m总和不超过500000）

思路：若不考虑深度限制，我们可以先给每个点赋点权1，再把每种颜色的节点按dfs序排序后相邻的节点的LCA的点权减1，每次求子树和即可回答询问，若有限制深度，我们可以按深度顺序把点一个个加入树中，每种颜色用个set维护，强制在线只要把求子树和的线段树可持久化就可以了，总复杂度O(nlogn)。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<set>
using namespace std;
inline int read()
{
    int x;char c;
    while((c=getchar())<''||c>'');
    for(x=c-'';(c=getchar())>=''&&c<='';)x=(x<<)+(x<<)+c-'';
    return x;
}
#define MN 100000
#define K 17
#define ND 8000000
struct edge{int nx,t;}e[MN+];
int h[MN+],en,c[MN+],d[MN+],l[MN+],r[MN+],cnt,fa[K][MN+],p[MN+],rt[MN+],tn;
class stcmp{public:inline bool operator()(int a,int b){return l[a]<l[b];}};
set<int,stcmp> st[MN+];
set<int,stcmp>::iterator x,y,z;
struct node{int l,r,s;}t[ND+];
int renew(int x,int l,int r,int k,int z)
{
    int p=++tn,mid=l+r>>;
    if(l==r)return t[p]=(node){,,t[x].s+z},p;
    if(k<=mid)return t[p]=(node){renew(t[x].l,l,mid,k,z),t[x].r,t[x].s+z},p;
    return t[p]=(node){t[x].l,renew(t[x].r,mid+,r,k,z),t[x].s+z},p;
}
int query(int x,int l,int r,int L,int R)
{
    if(l==L&&r==R)return t[x].s;
    int mid=l+r>>;
    if(R<=mid)return query(t[x].l,l,mid,L,R);
    if(L>mid)return query(t[x].r,mid+,r,L,R);
    return query(t[x].l,l,mid,L,mid)+query(t[x].r,mid+,r,mid+,R);
}
inline void ins(int x,int y){e[++en]=(edge){h[x],y};h[x]=en;}
void dfs(int x)
{
    l[x]=++cnt;
    for(int i=h[x];i;i=e[i].nx)d[e[i].t]=d[x]+,dfs(e[i].t);
    r[x]=cnt;
}
bool cmp(int a,int b){return d[a]<d[b];}
int lca(int x,int y)
{
    int dx=d[x]-d[y],i;
    if(dx<)dx=-dx,swap(x,y);
    for(i=;dx;dx>>=,++i)if(dx&)x=fa[i][x];
    if(x==y)return x;
    for(i=K;i--;)if(fa[i][x]!=fa[i][y])x=fa[i][x],y=fa[i][y];
    return fa[][x];
}
int main()
{
    int T,n,m,i,j,t;
    for(T=read();T--;)
    {
        n=read();m=read();
        for(i=;i<=n;++i)c[i]=read(),p[i]=i,st[i].clear();
        memset(h,t=tn=en=cnt=,sizeof(int)*(n+));
        for(i=;i<=n;++i)ins(fa[][i]=read(),i);
        for(i=;i<K;++i)for(j=;j<=n;++j)fa[i][j]=fa[i-][fa[i-][j]];
        dfs();sort(p+,p+n+,cmp);
        for(i=;i<=n;++i)
        {
            t=renew(t,,n,l[p[i]],);
            st[c[p[i]]].insert(p[i]);
            x=y=z=st[c[p[i]]].find(p[i]);--x;++z;
            if(y!=st[c[p[i]]].begin()&&z!=st[c[p[i]]].end())t=renew(t,,n,l[lca(*x,*z)],);
            if(y!=st[c[p[i]]].begin())t=renew(t,,n,l[lca(*x,*y)],-);
            if(z!=st[c[p[i]]].end())t=renew(t,,n,l[lca(*y,*z)],-);
            rt[d[p[i]]]=t;
        }
        for(t=;m--;)
        {
            i=read()^t;j=read()^t;
            printf("%d\n",t=query(rt[d[i]+j>d[p[n]]?d[p[n]]:d[i]+j],,n,l[i],r[i]));
        }
    }
}