題意:給一棵樹,每個點有一個名字,有若干個詢問,每次詢問給出,,表示子樹中與之深度差爲的點的個數,如果點名字相同則只計算一次。
用維護每個深度的名字出現情況。
好像被卡了常數。
#include <bits/stdc++.h>
#define regi register int
int n;
int m;
int head[500000],tot;
int lsans[500000];
int size[500000];
int deep[500000];
int son[500000];
char user_name[500000];
unsigned int id[500000];
std::vector<int>wait_solve[500000],ans_id[500000];
std::map<std::pair<int,unsigned int>,int>parent_map;
int ans[500000];
struct edge{
int to;
int nxt;
}e[500000];
inline void add(int x,int y){
e[++tot]={y,head[x]};
head[x]=tot;
}
void dfs(int x,int f=0){
size[x]=1;
deep[x]=deep[f]+1;
for(regi i=head[x];i;i=e[i].nxt){
regi y=e[i].to;
if(y==f)
continue;
dfs(y,x);
size[x]+=size[y];
if(size[son[x]]<size[y])
son[x]=y;
}
}
void Calc(int x,int f,int heavy){
lsans[deep[x]]+=(++parent_map[std::make_pair(deep[x],id[x])]==1);
for(regi i=head[x];i;i=e[i].nxt){
regi y=e[i].to;
if(y==f||y==heavy)
continue;
Calc(y,x,heavy);
}
}
void Clear(int x,int f){
lsans[deep[x]]-=(--parent_map[std::make_pair(deep[x],id[x])]==0);
for(regi i=head[x];i;i=e[i].nxt){
regi y=e[i].to;
if(y==f)
continue;
Clear(y,x);
}
}
void dsu(int x,int f=-1){
for(regi i=head[x];i;i=e[i].nxt){
regi y=e[i].to;
if(y==f||y==son[x])
continue;
dsu(y,x);
Clear(y,x);
}
if(son[x])
dsu(son[x],x);
Calc(x,f,son[x]);
for(regi i=0;i<wait_solve[x].size();++i)
ans[ans_id[x][i]]=lsans[deep[x]+wait_solve[x][i]];
}
main(){
scanf("%d",&n);
for(regi i=1,x;i<=n;++i){
scanf("%s%d",user_name+1,&x);
for(regi j=1,limit=strlen(user_name+1);j<=limit;++j)
id[i]=id[i]*1331+user_name[j]-'a'+1;
add(i,x);
add(x,i);
}
dfs(0);
scanf("%d",&m);
for(regi i=1,x,y;i<=m;++i){
scanf("%d%d",&x,&y);
if(y>n)
continue;
wait_solve[x].push_back(y);
ans_id[x].push_back(i);
}
dsu(0);
for(regi i=1;i<=m;++i)
printf("%d\n",ans[i]);
return 0;
}