洛谷-Blood Cousins Return-(树上启发式合并+set)

DIV2-E

题意：
就是给你一个森林，然后每个节点有一个名字，然后让你输出每个节点代表的子树内，且深度是该节点+k的节点的名字不同的有多少个。

思考：
很明显了，离线查询每个子树，一般就是启发式。这个题呢，说的是森林其实也一样，每个根节点都跑一遍就行了。对于某个深度有多少名字，直接维护n个深度，把名字放在set里就行了。

代码：

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);

using namespace std;
const int mod = 1e9+7,inf = 1e18;
const int N = 2e5+10,M = 2010;

int T,n,m,k;
string va[N];
int siz[N],son[N],dep[N],acc[N];
int anw[N];

set<string > s[N];
vector<PII > v[N];
vector<int > e[N];

void get(int now,int p)
{
	siz[now] = 1;
	dep[now] = dep[p]+1;
	for(auto spot:e[now])
	{
		if(spot==p) continue;
		get(spot,now);
		siz[now] += siz[spot];
		if(siz[spot]>siz[son[now]]) son[now] = spot;
	}
}

void add(int now,int p,int value,int wson)
{
	if(value==1) s[dep[now]].insert(va[now]);
	else if(s[dep[now]].count(va[now])) s[dep[now]].erase(va[now]);
	for(auto spot:e[now])
	{
		if(spot==p||spot==wson) continue;
		add(spot,now,value,wson);
	}
}

void dfs(int now,int p,int del)
{
	for(auto spot:e[now])
	{
		if(spot==p||spot==son[now]) continue;
		dfs(spot,now,1);
	}
	if(son[now]) dfs(son[now],now,0);
	add(now,p,1,son[now]);
	for(auto t:v[now]) anw[t.se] = s[dep[now]+t.fi].size();
	if(del) add(now,p,-1,0);
}

signed main()
{
	IOS;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		int a;
		cin>>va[i]>>a;
		e[a].pb(i);
		acc[i] = a;
	}
	cin>>m;
	for(int i=1;i<=m;i++)
	{
		int a,b;
		cin>>a>>b;
		v[a].pb({b,i});
	}
	for(int i=1;i<=n;i++)
	{
		if(!acc[i]) get(i,0),dfs(i,0,1); //如果是多个树的话，搜的时候dfs默认也是删掉的，要不然会对后面的树有影响。
	}
	for(int i=1;i<=m;i++) cout<<anw[i]<<"\n";
	return 0;
}

总结：
多多思考呀。