CF 246E. Blood Cousins Return [dsu on tree STL]

题意：
一个森林，求k级后代中多少种不同的权值

---

用set维护每个深度出现的权值
一开始一直在想删除怎么办，后来发现因为当前全局维护的东西里都是当前子树里的，如果要删除那么当前一定是轻儿子，直接清空set就行了没关系不会把一些要保留的也删除了
哈希直接上map

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <string>
#include <set>
#include <map>
using namespace std;
typedef long long ll;
#define pii pair<int, ll>
#define MP make_pair 
#define fir first
#define sec second
const int N=1e5+5;
int read(){
    char c=getchar();int x=0,f=1;
    while(c<'0'||c>'9'){if(c=='-')f=-1; c=getchar();}
    while(c>='0'&&c<='9'){x=x*10+c-'0'; c=getchar();}
    return x*f;
}

int n, a[N], Q, x, root[N];
char ch[30];
map<string, int> Hash; int m;
int id[N];
set<int> Set[N];
vector<pii> q[N];
int ans[N];
struct edge{int v, ne;}e[N<<1];
int cnt, h[N];
inline void ins(int u, int v) {
    e[++cnt]=(edge){v, h[u]}; h[u]=cnt;
}
int size[N], mx[N], deep[N], big[N];
void dfs(int u) {
    size[u]=1;
    for(int i=h[u];i;i=e[i].ne) {
        deep[e[i].v] = deep[u]+1;
        dfs(e[i].v);
        size[u] += size[e[i].v];
        if(size[e[i].v] > size[mx[u]]) mx[u] = e[i].v;
    }
}

pii f[N];
void update(int u, int val) {
    if(val == 1) Set[ deep[u] ].insert(id[u]);
    else Set[ deep[u] ].erase(id[u]);
    for(int i=h[u];i;i=e[i].ne) if(!big[e[i].v]) update(e[i].v, val);
}
inline int cal(int d) { return d>n ? 0 : Set[d].size(); }
void dfs(int u, int keep) {
    for(int i=h[u];i;i=e[i].ne) 
        if(e[i].v != mx[u]) dfs(e[i].v, 0);
    if(mx[u]) dfs(mx[u], 1), big[mx[u]]=1;
    update(u, 1);
    for(int i=0; i<(int)q[u].size(); i++) ans[q[u][i].fir] = cal(deep[u] + q[u][i].sec);
    big[mx[u]]=0;
    if(!keep) update(u, -1);
}

int main() {
    //freopen("in","r",stdin);
    n=read();
    for(int i=1; i<=n; i++) {
        scanf("%s",ch+1); x=read();
        string str(ch+1);
        if(Hash.count(str)) id[i] = Hash[str];
        else Hash[str] = id[i] = ++m;
        if(x == 0) root[++root[0]] = i;
        else ins(x, i);
    }
    //for(int i=1; i<=n; i++) printf("hi %d\n",id[i]);
    Q=read();
    for(int i=1; i<=Q; i++) x=read(), q[x].push_back(MP(i, read()));
    for(int i=1; i<=root[0]; i++) dfs(root[i]), dfs(root[i], 0);
    for(int i=1; i<=Q; i++) printf("%d\n",ans[i]);
}