[雜題]color Solution

題意:給一棵樹,每個點有一個名字,有若干個詢問,每次詢問給出xxyy,表示xx子樹中與之深度差爲yy的點的個數,如果點名字相同則只計算一次。
dsu on treedsu~on~tree
mapmap維護每個深度的名字出現情況。
好像被卡了常數。
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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章