【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]);
}
}