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 射手座之日

代碼

總結

以上是生活随笔為你收集整理的树上启发式合并 简单例题的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。