大意是每個(gè)點(diǎn)有一個(gè)顏色，詢問以每個(gè)點(diǎn)為根的子樹中出現(xiàn)次數(shù)最多的顏色是誰(shuí)，如果有多個(gè)輸出編號(hào)和。

這個(gè)東西用數(shù)據(jù)結(jié)構(gòu)是不太好做的，考慮優(yōu)化暴力。

首先最裸的的暴力就是在dfs序上跑莫隊(duì)，但因?yàn)槭菢涞慕Y(jié)構(gòu)所以有更優(yōu)美的暴力。

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

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

0.首先輕重鏈剖分。

1.然后先處理輕兒子，處理完把貢獻(xiàn)刪除。

2.處理重兒子，保留貢獻(xiàn)。

3.把輕兒子的貢獻(xiàn)再加上。

4.算這個(gè)點(diǎn)的答案。

5.如果這個(gè)點(diǎn)是輕兒子就把數(shù)組里的東西清空。

因?yàn)槊總€(gè)點(diǎn)到根只有$log$條輕鏈，所以在執(zhí)行3和5時(shí)每個(gè)點(diǎn)最多加入刪除$log$次。

復(fù)雜度$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; }

　　

?

轉(zhuǎn)載于:https://www.cnblogs.com/ezyzy/p/7045169.html

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。