洛谷 P1352 没有上司的舞会

闲聊

某历史的梗:我的上司的上司,不是我的上司

上面这一条可能有些人会对这个题产生误会,我觉得题目描述如果全篇的“上司”改成“直接上司”的话,这个误会可能会消除吧。。。不过也没关系,毕竟样例推一下就知道了

正题

我们把一个人的所有准备去舞会的直接手下的开心值加起来,和这个人的开心值做比较,如果这个上司去更优,那么就让这个人去而这个人的手下不去了。

这个过程对其他的地方没有任何影响(也就是没有后效性),所以这个方法可行。

对于每一个人,我们要知道他的那些手下准备去舞会,于是我们可以从最底层开始往上推,判断当前是这个人的直接手下去还是这个人去,一直推到根节点。

这个过程从根节点写一个递归即可。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;
int n,u,v,cnt,head[100010],w[100010],p[100010],come[100010],ans;
struct EDGE{
	int ver,nxt;
}edge[100010];
void add(int u,int v){//链式前向星
	edge[++cnt].ver=v;
	edge[cnt].nxt=head[u];
	head[u]=cnt;
}
void dfs(int now){
	int sum=0,tmp;
	for (int i=head[now];i;i=edge[i].nxt){遍历这个人所有的边寻找手下
		tmp=edge[i].ver;dfs(tmp);
		if (come[tmp])sum+=w[tmp];
	}
	if (w[now]>sum){//这个人去更优
		come[now]=1;ans+=w[now];ans-=sum;
	}
}
int main(){
	scanf("%d",&n);
	for (int i=1;i<=n;i++){
		scanf("%d",&w[i]);
	}
	for (int i=1;i<n;i++){
		cin>>u>>v;add(v,u);p[u]=1;//我们是从领导向手下连边的
	}
	for (int i=1;i<=n;i++){
		if (!p[i]){//如果为根节点
			dfs(i);cout<<ans;return 0;
		}
	}
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值