Codeforces Round #151 (Div. 2) E. Blood Cousins Return

题目描述：戳这里

题解：

这题的大意是给出一颗树，每个节点有一个颜色，和m个询问，每个询问要求某一个节点vi的子树与它距离=ki 的节点中不同的颜色种类。

这里只涉及到对子树的统计并且不用修改，那么我们就可以用dsu on tree来解决。
具体可以看CF中这位dalao的解析
对于这题，我们要求不同的颜色种类，那么只要开一个set或者map存一下就好了。

代码如下：

#include<cstdio>
#include<string>
#include<iostream>
#include<map>
using namespace std;
const int maxn=1e5+5;
int n,m,tot,Son,col,lnk[maxn],son[maxn],nxt[maxn],maxson[maxn],c[maxn],siz[maxn],dep[maxn];
int tot1,lnkq[maxn],sonq[maxn],idq[maxn],nxtq[maxn],ans[maxn];
string s;
map<string,int> hsh1;
map<int,bool> hsh[maxn];
void add(int x,int y){son[++tot]=y,nxt[tot]=lnk[x],lnk[x]=tot;}
void addq(int x,int y,int z){sonq[++tot1]=y,idq[tot1]=z,nxtq[tot1]=lnkq[x],lnkq[x]=tot1;}
void dfs1(int x){
    siz[x]=1;
    for (int j=lnk[x];j;j=nxt[j]){
        dep[son[j]]=dep[x]+1;
        dfs1(son[j]);
        siz[x]+=siz[son[j]];
        if (siz[son[j]]>siz[maxson[x]]) maxson[x]=son[j];
    }
}
void calc(int x,int op){
    if (op==1) hsh[dep[x]][c[x]]=1;
    else hsh[dep[x]].clear();
    for (int j=lnk[x];j;j=nxt[j])
    if (son[j]!=Son) calc(son[j],op);
}
void dfs(int x,bool pd){
    for (int j=lnk[x];j;j=nxt[j])
    if (son[j]!=maxson[x]) dfs(son[j],0);
    if (maxson[x]) dfs(maxson[x],1),Son=maxson[x];
    calc(x,1); Son=0;
    for (int j=lnkq[x];j;j=nxtq[j]) {
        int h=dep[x]+sonq[j];
        if (h>n) continue;
        ans[idq[j]]=hsh[h].size();
    }
    if (!pd) calc(x,-1);
}
int main(){
    scanf("%d",&n);
    for (int i=1;i<=n;i++) {
        cin>>s; int x;scanf("%d",&x);
        add(x,i);
        if (hsh1[s]==0) hsh1[s]=++col;
        c[i]=hsh1[s];
    }
    scanf("%d",&m);
    for (int i=1;i<=m;i++) {
        int x,y; scanf("%d %d",&x,&y);
        addq(x,y,i);
    }
    for (int j=lnk[0];j;j=nxt[j]) dfs1(son[j]);
    for (int j=lnk[0];j;j=nxt[j]) dfs(son[j],0);
    for (int i=1;i<=m;i++) printf("%d\n",ans[i]);
    return 0;
}