246E.Bloods Counsin Return（离线+树上启发式合并）

给出一棵树，每个节点上有一个字符串。

每次询问子树x内的第y层儿子中不重复名字的个数。

题解：

树上启发式合并的过程中，数组f维护第i层中有多少种不同的字符串。

这里不太好直接维护，用set维护f数组即可。

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+100;
int c[maxn];
vector<int> g[maxn];
set<string> f[maxn];
int L[maxn],R[maxn],id[maxn];
int dep[maxn];
int size[maxn];
int tot;
int son[maxn];
int n,m;
int ans[maxn];
vector<pair<int,int> > q[maxn];
string s[maxn];

void dfs1 (int x,int pre) {
	dep[x]=dep[pre]+1;
	size[x]=1;
	L[x]=++tot;
	id[tot]=x;
	for (int y:g[x]) {
		if (y==pre) continue;
		dfs1(y,x);
		size[x]+=size[y];
		if (size[son[x]]<size[y]) son[x]=y;
	}
	R[x]=tot;
}
void cal (int x,int pre) {
	f[dep[x]].insert(s[x]);
	for (int y:g[x]) {
		if (y==son[x]||y==pre) continue;
		for (int j=L[y];j<=R[y];j++) {
			int z=id[j];
			f[dep[z]].insert(s[z]);
		}
	}
	for (pair<int,int> y:q[x]) {
		ans[y.second]=f[y.first].size();
	}
}
void dfs2 (int x,int pre,int kp) {
	for (int y:g[x]) {
		if (y==son[x]||y==pre) continue;
		dfs2(y,x,0);
	}
	if (son[x]) {
		dfs2(son[x],x,1);
	}
	cal(x,pre); 
	if (!kp) for (int i=L[x];i<=R[x];i++) f[dep[id[i]]].clear();
}
int rt;
int main () {
	scanf("%d",&n);
	for (int i=1;i<=n;i++) {
		int x;
		cin>>s[i]>>x;
		g[x].push_back(i);
	}
	
	scanf("%d",&m);
	dfs1(0,0);
	//for (int i=1;i<=n;i++) printf("%d ",dep[i]);
	//printf("
");
	for (int i=1;i<=m;i++) {
		int x,y;
		scanf("%d%d",&x,&y);
		//printf("%d %d

",dep[x],y);
		q[x].push_back(make_pair(dep[x]+y,i));
	} 
	dfs2(0,-1,1);
	for (int i=1;i<=m;i++) printf("%d
",ans[i]);
}