E. Lomsat gelral
题意:
输入
n
(
1
e
5
)
n(1e5)
n(1e5)
输入
c
1
,
c
2
,
…
,
c
n
c_1,c_2,\dots,c_n
c1,c2,…,cn表示节点颜色
接下来
n
−
1
n-1
n−1行,每行输入
u
,
v
u,v
u,v,表示树边。
注意 树根为1
求对于每一个节点所对应的子树,求出现颜色次数最多的贡献。
题解:
直接上dsu就好了
代码:
#pragma GCC optimize(2)
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+9;
#define ll long long
ll c[N],n,ans[N];
struct Edge{
int v,nxt;
}e[N<<2];
int head[N],cnt;
inline void add(int u,int v){
e[cnt]=(Edge){v,head[u]};
head[u]=cnt++;
}
int son[N],siz[N];
void dfs(int u,int fa){
siz[u]=1;
for(int i=head[u];~i;i=e[i].nxt){
Edge e1=e[i];
if(e1.v==fa)continue;
dfs(e1.v,u);//这句话忘写了,超时卡了好久
siz[u]+=siz[e1.v];
if(siz[son[u]]<siz[e1.v])son[u]=e1.v;
}
}
ll nowson,gs[N],sum,mx;
void calc(int u,int fa,int val){
gs[c[u]]+=val;
int t=gs[c[u]];
if(t>mx)mx=t,sum=c[u];
else if(t==mx)sum+=(long long)c[u];
for(int i=head[u];~i;i=e[i].nxt){
Edge e1=e[i];
if(e1.v==fa||e1.v==nowson)continue;
calc(e1.v,u,val);
}
}
void dsu(int u,int fa,int kep){
for(int i=head[u];~i;i=e[i].nxt){
Edge e1=e[i];
if(e1.v==fa||e1.v==son[u])continue;
dsu(e1.v,u,0);
}
if(son[u])dsu(son[u],u,1),nowson=son[u];
calc(u,fa,1),nowson=0;
ans[u]=sum;
if(!kep){
calc(u,fa,-1);
mx=0;
}
}
signed main(){
// freopen("tt.in","r",stdin),freopen("tt.out","w",stdout);
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
cin>>n;
for(int i=1;i<=n;i++)cin>>c[i];
memset(head,-1,sizeof(head));
for(int i=1,u,v;i<n;i++){
cin>>u>>v;
add(u,v),add(v,u);
}
dfs(1,0);
dsu(1,0,1);
for(int i=1;i<=n;i++)cout<<ans[i]<<" ";
cout<<endl;
return 0;
}