【DSU ON TREE】CF246E Blood Cousins Return(对点)

【DSU ON TREE】CF246E Blood Cousins Return(对点)

题意：

给定一片森林，每次询问一个节点的K-Son共有个多少不同的名字。一个节点的K-Son即为深度是该节点深度加K的节点。

思路：

用map存储更新，注意是森林

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxn=1e5+5;
const ll mod=998244353;
unordered_map<int,unordered_map<string,int>>tmp;
int t,n,m,k1,k,tim,num,sum;
int dui[maxn],dfn[maxn],siz[maxn],vis[maxn],id[maxn],ans[maxn],a[maxn],son[maxn],dep[maxn];
char h[100005][25];
struct node{
	int id,dep;
};
vector<node>que[maxn];
vector<int>mp[maxn];
void dfs1(int x){//dp求重心 
	siz[x]=1;
	dfn[x]=++tim;
	id[tim]=x;
	for(int i=0;i<mp[x].size();i++){
		int y=mp[x][i];
		dep[y]=dep[x]+1;
		dfs1(y);
		siz[x]+=siz[y];
		if(siz[y]>siz[son[x]])son[x]=y;
	}
}
void dfs(int x,int op){
	for(int i=0;i<mp[x].size();i++){
		int y=mp[x][i];
		if(y!=son[x])dfs(y,0);
	}
	if(son[x])dfs(son[x],1);
	for(int i=0;i<mp[x].size();i++){
		int v=mp[x][i];
		if(v==son[x])continue;
        for(int j=dfn[v];j<=dfn[v]+siz[v]-1;j++){
        	int y=id[j];
        	string p(h[y],h[y]+strlen(h[y]));
        	tmp[dep[y]][p]++;
		}
	}
	string p(h[x],h[x]+strlen(h[x]));
	tmp[dep[x]][p]++;
	for(int i=0;i<que[x].size();i++){
		ans[que[x][i].id]=tmp[que[x][i].dep].size();
	}
	if(!op)tmp.clear();
} 
int main(){
	scanf("%d",&n);
	for(int i=1,x;i<=n;i++){
		scanf("%s%d",h[i],&x);
		if(x==0)dui[++t]=i;
		else mp[x].push_back(i);
	}
	for(int i=1;i<=t;i++)dfs1(dui[i]);
	scanf("%d",&m);
	for(int i=1,x,y;i<=m;i++){
		scanf("%d%d",&x,&y);
		que[x].push_back({i,dep[x]+y});
	}
	for(int i=1;i<=t;i++)dfs(dui[i],0);
	for(int i=1;i<=m;i++){
		printf("%d\n",ans[i]);
	}
}