[杂题]color Solution

题意:给一棵树,每个点有一个名字,有若干个询问,每次询问给出xxxyyy,表示xxx子树中与之深度差为yyy的点的个数,如果点名字相同则只计算一次。
dsu on treedsu~on~treedsu on tree
mapmapmap维护每个深度的名字出现情况。
好像被卡了常数。
code:code:code:

#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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值