U41492 树上数颜色
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <set>
#include <vector>
#include <queue>
using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
typedef pair<LL,int> PLI;
const int N = 1e5+5,mod=1e9+7;int h[N],ne[N<<1],e[N<<1],idx=0;
void add(int a,int b){e[idx]=b,ne[idx]=h[a],h[a]=idx++;return ;}
int co[N],son[N],siz[N],cnt[N],ans[N],sum,flag;
// flag 是暴力统计的时候 重儿子也要遍历 但是根节点的重儿子不遍历 void dfs(int u,int fa)
{siz[u]++;for(int i=h[u];~i;i=ne[i]){if(e[i]==fa)continue;dfs(e[i],u);if(siz[e[i]]>siz[son[u]])son[u]=e[i];siz[u]+=siz[e[i]];}return ;
}void calc(int u,int fa,int val)
{if(val==1){if(!cnt[co[u]])sum++;cnt[co[u]]++;}else {cnt[co[u]]--;if(!cnt[co[u]])sum--;}for(int i=h[u];~i;i=ne[i])if(e[i]!=fa&&e[i]!=flag)calc(e[i],u,val);
}void dfs(int u,int fa,int keep)
{for(int i=h[u];~i;i=ne[i])if(e[i]!=fa&&e[i]!=son[u])dfs(e[i],u,0);if(son[u])dfs(son[u],u,1),flag=son[u];// 代表calc的时侯u节点的重儿子不遍历 calc(u,fa,1); flag=0; ans[u]=sum;if(!keep)calc(u,fa,-1),sum=0;
}int main()
{int n;scanf("%d",&n);for(int i=1;i<=n;i++)h[i]=-1;for(int i=1,a,b;i<n;i++)scanf("%d%d",&a,&b),add(a,b),add(b,a);for(int i=1;i<=n;i++)scanf("%d",co+i);dfs(1,0);dfs(1,0,1);for(int i=1;i<=n;i++)cout<<ans[i]<<endl;return 0;
}
CF600E Lomsat gelral
代码
CF570D Tree Requests
代码
CF208E Blood Cousins
代码
P4149 [IOI2011]Race
代码
P4556 [Vani有约会]雨天的尾巴 /【模板】线段树合并
代码
U95602 射手座之日
代码