题面：给定一棵n个点的树,每个点有一个颜色。

问对于每一条边，删除这条边之后形成的两棵子树的颜色交集的大小。

数据范围：2<=n<=1e5, 1<=c[i]<=n

代码（dsu on tree）：

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+9;
int n,q,c[N],cnt[N],tot[N],in[N],out[N],id[N],dfn;
int sz[N],son[N],sid[N];
struct edge{int to,id;};
vector<edge> e[N];
int ans[N];
void dfs1(int x,int fa){
	sz[x]=1,in[x]=++dfn,id[dfn]=x;
	for(int i=0;i<e[x].size();i++){
		int v=e[x][i].to;
		if(v==fa) continue;
		dfs1(v,x);
		sz[x]+=sz[v];
		if(sz[e[x][i].to]>sz[son[x]]) son[x]=e[x][i].to,sid[x]=e[x][i].id;
	}
	out[x]=dfn;
}
int res;
void mod(int x,bool tag){
	assert(c[x]<N);
	if(tag){
		cnt[c[x]]++;
		if(cnt[c[x]]==1) res++;
		if(cnt[c[x]]==tot[c[x]]) res--;
	}
	else cnt[c[x]]--;
}
bool solvebf(int x,bool tag){
	for(int i=in[x];i<=out[x];i++) mod(id[i],tag);
}
void dfs2(int x,int fa,int eid,bool h){
	for(int i=0;i<e[x].size();i++){
		int v=e[x][i].to;
		if(v==fa||v==son[x]) continue;
		dfs2(v,x,e[x][i].id,0);
	}
	if(son[x]) dfs2(son[x],x,sid[x],1);
	for(int i=0;i<e[x].size();i++){
		int v=e[x][i].to;
		if(v==fa||v==son[x]) continue;
		solvebf(v,1);
	}
	mod(x,1);
	ans[eid]=res;
	if(!h) solvebf(x,0),res=0;
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++) cin>>c[i],tot[c[i]]++; 
	for(int i=1;i<n;i++){
		int u,v;
		cin>>u>>v;
		e[u].push_back((edge){v,i});
		e[v].push_back((edge){u,i});
	}
	dfs1(1,0);
	dfs2(1,0,0,0);
	for(int i=1;i<n;i++) cout<<ans[i]<<endl;
	return 0;
}

报错：killed-segmentation fault

发现即使是n<=1000的小数据也会RE，注释掉solvebf函数后就不RE了，怀疑是数组越界以外的原因