标签:org print omsa const printf tree void pac ==
考虑一下子树的问题,我们可以把一棵树的dfn序搞出来,那么子树就是序列上的一段连续的区间.
然后就可以莫队飞速求解了.
但是这题还有\(\Theta(nlog_n)\)的做法.能有\(\Theta(n\sqrt{n})\)的做法要什么\(logn\)的
考虑\(dsu\ on\ tree\),与莫队没有任何区别.
如果不会的话,请自行跳转小Z的袜子并且切掉.
#include<stdio.h>
#include<stdlib.h>
#include<algorithm>
#include<iostream>
using namespace std;
const int N=100010;
int n,front[N],nxt[N<<1],to[N<<1],cnt,siz[N],son[N],num[N],c[N],Max,b[N];
long long ans[N];
pair<int,long long>f[N];
void Add(int u,int v){to[++cnt]=v;nxt[cnt]=front[u];front[u]=cnt;}
void dfs(int u,int fa){
siz[u]=1;
for(int i=front[u];i;i=nxt[i]){
int v=to[i];
if(v==fa)continue;
dfs(v,u);siz[u]+=siz[v];
if(siz[v]>siz[son[u]])son[u]=v;
}
}
void add(int u,int fa,int opt){
int &p=num[c[u]];
f[p].first--;f[p].second-=c[u];
p+=opt;
f[p].first++;f[p].second+=c[u];
if(opt==1)Max=max(Max,p);
else if(!f[Max].first)Max--;
for(int i=front[u];i;i=nxt[i])
{
int v=to[i];
if(v!=fa && !b[v])add(v,u,opt);
}
}
void dfs(int u,int fa,int opt){
for(int i=front[u];i;i=nxt[i]){
int v=to[i];
if(v!=fa && v!=son[u])dfs(v,u,0);
}
if(son[u])dfs(son[u],u,1);b[son[u]]=1;
add(u,fa,1);
ans[u]=f[Max].second;
b[son[u]]=0;
if(!opt)add(u,fa,-1);
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++)scanf("%d",&c[i]);
for(int i=1;i<n;i++){
int u,v;scanf("%d%d",&u,&v);
Add(u,v);Add(v,u);
}
dfs(1,1);dfs(1,1,1);
for(int i=1;i<=n;i++)printf("%lld%c",ans[i],i==n?'\n':' ');
return 0;
}
标签:org print omsa const printf tree void pac ==
原文地址:https://www.cnblogs.com/MLEAutomaton/p/10317941.html