CF 600 E Lomsat gelral —— 树上启发式合并
2024-09-03 13:21:19
题目:http://codeforces.com/contest/600/problem/E
看博客:https://blog.csdn.net/blue_kid/article/details/82192641
https://blog.csdn.net/clove_unique/article/details/60772212
还是不太明白复杂度...
代码如下:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
int const xn=1e5+;
int n,c[xn],hd[xn],ct,to[xn<<],nxt[xn<<],cnt[xn],siz[xn],son[xn],mx,big;
ll ans[xn],sum;
void add(int x,int y){to[++ct]=y; nxt[ct]=hd[x]; hd[x]=ct;}
int rd()
{
int ret=,f=; char ch=getchar();
while(ch<''||ch>''){if(ch=='-')f=; ch=getchar();}
while(ch>=''&&ch<='')ret=(ret<<)+(ret<<)+ch-'',ch=getchar();
return f?ret:-ret;
}
void dfs(int x,int fa)
{
siz[x]=;
for(int i=hd[x],u;i;i=nxt[i])
{
if((u=to[i])==fa)continue;
dfs(u,x); siz[x]+=siz[u];
if(siz[u]>siz[son[x]])son[x]=u;
}
}
void add(int x,int fa,int v)
{
cnt[c[x]]+=v;
if(cnt[c[x]]>mx)sum=c[x],mx=cnt[c[x]];
else if(cnt[c[x]]==mx)sum+=c[x];
for(int i=hd[x],u;i;i=nxt[i])
if((u=to[i])!=fa&&u!=big)add(u,x,v);
}
void dfs2(int x,int fa,int keep)
{
for(int i=hd[x],u;i;i=nxt[i])
if((u=to[i])!=fa&&u!=son[x])dfs2(u,x,);
if(son[x])dfs2(son[x],x,),big=son[x];
add(x,fa,); big=;
ans[x]=sum;
if(!keep)add(x,fa,-),mx=sum=;//=0!
}
int main()
{
n=rd();
for(int i=;i<=n;i++)c[i]=rd();
for(int i=,x,y;i<n;i++)
{
x=rd(); y=rd();
add(x,y); add(y,x);
}
dfs(,); dfs2(,,);
for(int i=;i<=n;i++)printf("%I64d ",ans[i]);
return ;
}
最新文章
- Application Request Route实现IIS Server Farms集群负载详解
- Le li&#233; &#224; la l&#233;g&#232;ret&#233; semblait &#234;tre et donc plus simple
- 总结:在MyEclipse中部署一个wap应用时需要配置的环境变量,我的JDK是安装在C盘,mysql安装在D盘,Tomcat解压在E盘,所以路径一定要看清楚哦,!
- React生命周期
- robots.txt文件没错,为何总提示封禁
- python 注意事项
- ssh通道技术
- POJ1651:Multiplication Puzzle(区间DP)
- UVa-Palindromes
- js原生设计模式——7原型模式之真正的原型模式——对象复制封装
- 简单的后台数据和前台数据交互.net
- CentOS7下安装MySQL的安装与配置(yum) (转)
- 机器学习算法与Python实践之(五)k均值聚类(k-means)
- Vector 特性
- Android 百分比布局库(percent-support-lib) 解析与扩展
- ASP.NET 教程汇总
- 分离式部署LNMP
- Linux Centos7安装最新anslib
- day 5,格式化输出,for,while, break,continue,列表
- python的os模块总结