Codeforces Round #527 (Div. 3) . F Tree with Maximum Cost

题目链接
题意:给你一棵树,让你找一个顶点 i i i,使得这个点的 d i s ( i , j ) a [ j ] \sum dis(i,j)*a[j] dis(i,j)a[j]最大。 d i s ( i , j ) dis(i,j) dis(i,j) i i i j j j的距离。
思路:题解还是好看啊 先从1开始搜,预处理出当1为根的时候的答案记为 r e s res res,递归的过程中假设当前节点为 i i i,那么 s u m [ i ] sum[i] sum[i]数组的意义就是:从以当前顶点为根的所有子树的所有节点的权值和记为 s u m [ i ] sum[i] sum[i]
预处理完成之后就是题解中神奇的re-rooting(换根)。 还是从1开始递归,假设当前根为 v v v,他的一个子节点为 t o to to,那么 v > t o v->to v>to就是一个换根过程,需要做一下转变,类似于莫队的转移 ,首先由于根的变化,之前v到 n o w now now的某个子节点的距离是 x + 1 x+1 x+1的话,那么 n o w now now为根的时候这个距离就是 x x x了。 r e s res res就要减去 s u m [ k ] , s u m [ v ] sum[k],sum[v] sum[k],sum[v]也要减去 s u m [ k ] sum[k] sum[k],相应当, t o n o w to经过now tonow才能到的节点的到根的距离就要加1, r e s s u m [ v ] , s u m [ k ] s u m [ v ] res就要加上sum[v],sum[k]也要加上sum[v] ressum[v],sum[k]sum[v],然后记得回溯就行了。

#include<bits/stdc++.h>

#define LL long long
#define fi first
#define se second
#define mp make_pair
#define pb push_back

using namespace std;

LL gcd(LL a,LL b){return b?gcd(b,a%b):a;}
LL lcm(LL a,LL b){return a/gcd(a,b)*b;}
LL powmod(LL a,LL b,LL MOD){LL ans=1;while(b){if(b%2)ans=ans*a%MOD;a=a*a%MOD;b/=2;}return ans;}

const int N =2e5+232;
int n,a[N];
vector<int>v[N];
LL sum[N],res,ans;
void dfs1(int now,int pre=-1,int h=0){
	LL ans=0;
	res+=1ll*h*a[now];
	sum[now]=a[now];
	for(auto k:v[now]){
		if(k==pre)continue;
		dfs1(k,now,h+1);
		sum[now]+=sum[k];
	}
}
void dfs2(int now,int p=-1){
	ans=max(ans,res);
	for(auto k:v[now]){
		if(k==p)continue;
		res-=sum[k];
		sum[now]-=sum[k];
		res+=sum[now];
		sum[k]+=sum[now];
		dfs2(k,now);
		sum[k]-=sum[now];
		res-=sum[now];
		sum[now]+=sum[k];
		res+=sum[k];
	}
}
int main(){
	ios::sync_with_stdio(false);
	cin>>n;
	for(int i=1;i<=n;i++)cin>>a[i];
	for(int i=1;i<n;i++){
		int s,t;
		cin>>s>>t;
		v[s].pb(t);
		v[t].pb(s);
	}
	dfs1(1);
	dfs2(1);
	cout<<ans;
	return 0;
}
全部评论

相关推荐

避坑恶心到我了大家好,今天我想跟大家聊聊我在成都千子成智能科技有限公司(以下简称千子成)的求职经历,希望能给大家一些参考。千子成的母公司是“同创主悦”,主要经营各种产品,比如菜刀、POS机、电话卡等等。听起来是不是有点像地推销售公司?没错,就是那种类型的公司。我当时刚毕业,急需一份临时工作,所以在BOSS上看到了千子成的招聘信息。他们承诺无责底薪5000元,还包住宿,这吸引了我。面试的时候,HR也说了同样的话,感觉挺靠谱的。于是,我满怀期待地等待结果。结果出来后,我通过了面试,第二天就收到了试岗通知。试岗的内容就是地推销售,公司划定一个区域,然后你就得见人就问,问店铺、问路人,一直问到他们有意向为止。如果他们有兴趣,你就得摇同事帮忙推动,促进成交。说说一天的工作安排吧。工作时间是从早上8:30到晚上18:30。早上7点有人叫你起床,收拾后去公司,然后唱歌跳舞(销售公司都这样),7:55早课(类似宣誓),8:05同事间联系销售话术,8:15分享销售技巧,8:30经理训话。9:20左右从公司下市场,公交、地铁、自行车自费。到了市场大概10点左右,开始地推工作。中午吃饭时间大约是12:00,公司附近的路边盖饭面馆店自费AA,吃饭时间大约40分钟左右。吃完饭后继续地推工作,没有所谓的固定中午午休时间。下午6点下班后返回公司,不能直接下班,需要与同事交流话术,经理讲话洗脑。正常情况下9点下班。整个上班的一天中,早上到公司就是站着的,到晚上下班前都是站着。每天步数2万步以上。公司员工没有自己的工位,百来号人挤在一个20平方米的空间里听经理洗脑。白天就在市场上奔波,公司的投入成本几乎只有租金和工资,没有中央空调。早上2小时,晚上加班2小时,纯蒸桑拿。没有任何福利,节假日也没有3倍工资之类的。偶尔会有冲的酸梅汤和西瓜什么的。公司的晋升路径也很有意思:新人—组长—领队—主管—副经理—经理。要求是业绩和团队人数,类似传销模式,把人留下来。新人不能加微信、不能吐槽公司、不能有负面情绪、不能谈恋爱、不能说累。在公司没有任何坐的地方,不能依墙而坐。早上吃早饭在公司外面的安全通道,未到上班时间还会让你吃快些不能磨蹭。总之就是想榨干你。复试的时候,带你的师傅会给你营造一个钱多事少离家近的工作氛围,吹嘘工资有多高、还能吹自己毕业于好大学。然后让你早点来公司、无偿加班、抓住你可能不会走的心思进一步压榨你。总之,大家在找工作的时候一定要擦亮眼睛,避免踩坑!———来自网友
qq乃乃好喝到咩噗茶:不要做没有专业门槛的工作
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务