闲聊
某历史的梗:我的上司的上司,不是我的上司
上面这一条可能有些人会对这个题产生误会,我觉得题目描述如果全篇的“上司”改成“直接上司”的话,这个误会可能会消除吧。。。不过也没关系,毕竟样例推一下就知道了
正题
我们把一个人的所有准备去舞会的直接手下的开心值加起来,和这个人的开心值做比较,如果这个上司去更优,那么就让这个人去而这个人的手下不去了。
这个过程对其他的地方没有任何影响(也就是没有后效性),所以这个方法可行。
对于每一个人,我们要知道他的那些手下准备去舞会,于是我们可以从最底层开始往上推,判断当前是这个人的直接手下去还是这个人去,一直推到根节点。
这个过程从根节点写一个递归即可。
#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;
}
}
}