dsu on tree
先分轻重儿子
先处理轻边,再处理重儿子
再加上轻儿子的答案

#include<iostream>
#include<cstdio>
#include<queue>
#include<algorithm>
#include<cmath>
#include<ctime>
#include<set>
#include<map>
#include<stack>
#include<cstring>
#pragma GCC optimize(2)
#define inf 2147483647
#define ls rt<<1
#define rs rt<<1|1
#define lson ls,nl,mid,l,r
#define rson rs,mid+1,nr,l,r
#define N 1000010
#define For(i,a,b) for(long long i=a;i<=b;++i)
#define p(a) putchar(a)
#define g() getchar() using namespace std;
long long n,col[N],son[N],size[N],cnt[N],Max,Son;
long long sum,ans[N],x,y; struct node{
long long n;
node *next;
}*e[N]; void in(long long &x){
long long y=;char c=g();x=;
while(c<''||c>''){if(c=='-')y=-;c=g();}
while(c<=''&&c>=''){ x=(x<<)+(x<<)+c-'';c=g();}
x*=y;
}
void o(long long x){
if(x<){p('-');x=-x;}
if(x>)o(x/);
p(x%+'');
} void push(long long x,long long y){
node *p;
p=new node();
p->n=y;
if(e[x]==)
e[x]=p;
else{
p->next=e[x]->next;
e[x]->next=p;
}
} void dfs(long long x,long long fa){//重儿子
size[x]=;
for(node *i=e[x];i;i=i->next){
if(i->n==fa) continue;
dfs(i->n,x);
size[x]+=size[i->n];
if(size[i->n]>size[son[x]])
son[x]=i->n;
}
} void add(long long x,long long fa,long long val){
cnt[col[x]]+=val;
if(cnt[col[x]]>Max){
Max=cnt[col[x]];
sum=col[x];
}
else
if(cnt[col[x]]==Max)
sum+=col[x];
for(node *i=e[x];i;i=i->next){
if(i->n==fa||i->n==Son) continue;
add(i->n,x,val);
}
} void dfs2(long long x,long long fa,long long opt){
for(node *i=e[x];i;i=i->next){
if(i->n==fa) continue;
if(i->n!=son[x]) dfs2(i->n,x,);
}
if(son[x]){
dfs2(son[x],x,);
Son=son[x];
}
add(x,fa,);
Son=;
ans[x]=sum;
if(!opt){
add(x,fa,-);
sum=;
Max=;
}
} int main(){
in(n);
For(i,,n)
in(col[i]);
For(i,,n-){
in(x);in(y);
push(x,y);
push(y,x);
}
dfs(,);
dfs2(,,);
For(i,,n)
o(ans[i]),p(' ');
return ;
}

最新文章

  1. Javascript中document.execCommand()的用法
  2. 初探 Ext JS 6 (sencha touch/ext升级版)
  3. 自定义View_2_关于自定义组合View
  4. spring注入简记
  5. Javascript验证用户输入URL地址是否正确
  6. 【Linux C中文函数手册】 字符串转换函数
  7. Java基础知识强化之集合框架笔记05:Collection集合的遍历
  8. PyConChina2016 北京站 献给Python开发者
  9. 2018 HDU多校第三场赛后补题
  10. Spring Boot Actuator [监控与管理]
  11. maven 项目pom文件引入lib下的jar包
  12. openjudge------ 日期的种类题目
  13. css实现文本溢出显示省略号
  14. Windows has encountered a critical problem and will restart automatically in one minute. Please save your work now
  15. C++中三种创建对象的方法【转】
  16. 安装PyInstaller打包python
  17. HDU1349 Minimum Inversion Number 2016-09-15 13:04 75人阅读 评论(0) 收藏
  18. 应用开发之WinForm环境
  19. hive2.1.1配置
  20. Leetcode注意

热门文章

  1. for循环和字典预习
  2. Go语言简介以及安装
  3. Maven远程仓库地址修改(精)
  4. AFO成功
  5. dart中的typedef &lt;函数别名&gt;
  6. 使用WebStorm上传本地项目到GitHub和GitLab
  7. arcmap分类标注问题
  8. 转: sizeof,总结
  9. SSL和TLS漏洞验证
  10. Java之RabbitMQ(一)与SpringBoot整合