大意是每個點有一個顏色，詢問以每個點為根的子樹中出現次數最多的顏色是誰，如果有多個輸出編號和。

這個東西用數據結構是不太好做的，考慮優化暴力。

首先最裸的的暴力就是在dfs序上跑莫隊，但因為是樹的結構所以有更優美的暴力。

參考的博客：http://www.cnblogs.com/candy99/p/dsuontree.html

類似樹上啟發式合并，叫做dsu on tree。

0.首先輕重鏈剖分。

1.然后先處理輕兒子，處理完把貢獻刪除。

2.處理重兒子，保留貢獻。

3.把輕兒子的貢獻再加上。

4.算這個點的答案。

5.如果這個點是輕兒子就把數組里的東西清空。

因為每個點到根只有$log$條輕鏈，所以在執行3和5時每個點最多加入刪除$log$次。

復雜度$nlogn$

#include<bits/stdc++.h> #define ll long long #define N 100005 using namespace std; int v[N],head[N],ver[N*2],nxt[N*2],tot; void add(int a,int b) {tot++;nxt[tot]=head[a];head[a]=tot;ver[tot]=b;return; } int n,size[N],son[N]; void dfs(int x,int f) {size[x]=1;for(int i=head[x];i;i=nxt[i]){if(ver[i]==f)continue;dfs(ver[i],x);size[x]+=size[ver[i]];if(size[ver[i]]>size[son[x]])son[x]=ver[i];} } int cnt[N]; ll sum[N],ans[N]; int mx; void ds(int x,int f,int z) {sum[cnt[v[x]]]-=v[x];cnt[v[x]]+=z;sum[cnt[v[x]]]+=v[x];if(cnt[v[x]]>mx)mx=cnt[v[x]];else if(sum[mx]==0)mx--;for(int i=head[x];i;i=nxt[i]){if(ver[i]==f)continue;ds(ver[i],x,z);} } void dffs(int x,int f,int op) {for(int i=head[x];i;i=nxt[i]){if(ver[i]==f||ver[i]==son[x])continue;dffs(ver[i],x,0);}if(son[x])dffs(son[x],x,1);sum[cnt[v[x]]]-=v[x];cnt[v[x]]++;sum[cnt[v[x]]]+=v[x];if(cnt[v[x]]>mx)mx=cnt[v[x]];for(int i=head[x];i;i=nxt[i]){if(ver[i]==f||ver[i]==son[x])continue;ds(ver[i],x,1);}ans[x]=sum[mx];if(!op)ds(x,f,-1); } int main() {scanf("%d",&n);for(int i=1;i<=n;i++)scanf("%d",&v[i]);int t1,t2;for(int i=1;i<n;i++){scanf("%d%d",&t1,&t2);add(t1,t2);add(t2,t1);}dfs(1,-1);dffs(1,-1,0);for(int i=1;i<=n;i++){printf("%I64d ",ans[i]);}return 0; }

　　

?

轉載于:https://www.cnblogs.com/ezyzy/p/7045169.html

總結

以上是生活随笔為你收集整理的[Coderforces600E] Lomsat gelral的全部內容，希望文章能夠幫你解決所遇到的問題。

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