树上数颜色(板子)

//#pragma GCC optimize(2)
//#pragma GCC optimize(3,"Ofast","inline")
#include<bits/stdc++.h>
#define int long long
#define fi first
#define se second
#define pb push_back
#define pii pair<int,int>
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
const int inf=2e18+100;
const int maxn=2e5+100;
int col[maxn],cnt[maxn];
int mson[maxn],son[maxn];//mson重子节点,son儿子节点数 
int ans[maxn],res;
vector<int>g[maxn];
void dfs(int x,int f)
{
	son[x]=1;
	for(auto it:g[x])
	{
		if(it==f)continue;
		dfs(it,x);
		son[x]+=son[it];
		if(son[it]>son[mson[x]])mson[x]=it;
	}
}
void go(int x,int f,int op)
{
	cnt[col[x]]+=op;
	if(op==1&&cnt[col[x]]==1)res++;
	//if(op==-1&&cnt[col[x]]==0)res--;
	for(auto it:g[x])
	{
		if(it==f)continue;
		go(it,x,op);
	}
}
void dfs2(int x,int f,bool del)
{
	for(auto it:g[x])
	{
		if(it==f||it==mson[x])continue;
		dfs2(it,x,1);
	}
	if(mson[x])dfs2(mson[x],x,0);
	cnt[col[x]]++;
	if(cnt[col[x]]==1)res++;
	for(auto it:g[x])
	{
		if(it==f||it==mson[x])continue;
		go(it,x,1);
	}
	ans[x]=res;
	if(del)
	{
		res=0;//删除轻子节点信息 
		go(x,f,-1);
	}
}
signed main()
{
	IOS 
	int n,m;
	cin>>n;
	for(int i=1;i<n;i++)
	{
		int u,v;
		cin>>u>>v;
		g[u].pb(v);
		g[v].pb(u);
	}
	for(int i=1;i<=n;i++)
	{
		cin>>col[i];
	}
	dfs(1,-1);
	dfs2(1,-1,0);
	cin>>m;
	while(m--)
	{
		int x;
		cin>>x;
		cout<<ans[x]<<"\n";
	}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

thusloop

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值