Codeforces 1083A

给一棵树,每个节点、每条边都有权值,到一个会加上对应的权值,经过一条边会减去边上的权值,问从节点出发,最多能剩下多少权值。
树形dp,对于一个节点u,不可能又算它的父亲节点,又算它的两个两个(或以上)的儿子,原因是路径无法重复走。
dp[u]表示u到它的一个叶子节点的最大剩余的权值。
dp[u] = max(val[u] + dp[v] - cost(u,v)) v是u的孩子
更新答案时,首先要判断一下如果一个节点什么边也不走是否能构成最优解,然后对于u和儿子 v,u可能已经和其他的孩子节点构成一个比 单独的u 更好的情况,即 用 dp[u] + dp[v] - cost(u,v) 更新ans

#include <bits/stdc++.h>
#define f first
#define s second
using namespace std;
typedef pair<int,int> PII;
typedef long long LL;
const int N = 3e5 + 10;

int n,u,v,w,val[N];
vector<PII> G[N];
LL ans,dp[N];
void dfs(int u,int fa){
	dp[u] = (LL)val[u];
	ans = max(ans,dp[u]);
	for(int i = 0;i<G[u].size();i++){
		int v = G[u][i].f;
		if(v == fa) continue;
		dfs(v,u);
		ans = max(ans,dp[u] + dp[v] - G[u][i].s);
		dp[u] = max(dp[u],val[u] + dp[v] - G[u][i].s);
	}
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	cin>>n;
	for(int i = 1;i<=n;i++) cin>>val[i];
	for(int i = 0;i<n-1;i++){
		cin>>u>>v>>w;
		G[u].push_back({v,w});
		G[v].push_back({u,w});
	}
	dfs(1,-1);
	cout<<ans;
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值