#13 WA求调
查看原帖
#13 WA求调
1274736
Lijiangjun4楼主2024/7/27 22:45

#13 没过,哪位大神帮忙调一下

#include<iostream>
#include<set>
using namespace std;

int head[200005],nxt[200005],to[400005],cnt;
void add(int u,int v)
{
	++cnt;
	to[cnt]=v;
	nxt[cnt]=head[u];
	head[u]=cnt;
}


set<int> s[200005];
int c[200005],ans,n,k;
void dfs(int x,int fa)
{
	s[x].insert(c[x]);
	for(int i=head[x];i;i=nxt[i])
	{
		if(to[i]==fa) continue;
		dfs(to[i],x);
		if(s[x].size()<s[to[i]].size())
		{
			s[x].swap(s[to[i]]);
		}
	}
	for(int i=head[x];i;i=nxt[i])
	{
		if(to[i]==fa) continue;
		for(auto j:s[to[i]])
		{
			s[x].insert(j);
		}
	}
	if(s[x].size()>=k)
	{
		s[x].clear();
		ans++;
	}
}


int main()
{
	cin>>n>>k;
	for(int i=1;i<=n;i++) cin>>c[i];
	for(int i=1;i<=n-1;i++)
	{
		int x,y;
		cin>>x>>y;
		add(x,y);
		add(y,x);
	}
	dfs(1,0);
	cout<<ans;
	return 0;
}
2024/7/27 22:45
加载中...