[CodeForces-1084D][补题][树形DP(裸)]

题目是说有很多个城市,到达每个城市都可以获得相应的燃油,然后去往另外一个城市要消耗相应的燃油,不限制起点或者终点,怎么样才能获得最多的燃油?

题目第一行先输入一个n代表城市个数,然后接下来一行代表每个城市的获得燃油量,然后n-1行的路径和耗油量。

这道题使用树形DP算法,dfs搜索,一开始比较难以理解的就是怎么实现截取中间的一段路,因为dfs是从起始点一路向下搜索,后来发现,每次访问到一个点,先用这个城市的获得燃油量初始化dp数组,然后后面在更新dp的时候,比较子树+这个点的获得燃油量 - 消耗量(即拓展子树)和这个点初始化的权值,就可以决定是否保留子树。

注意,树形dp是使用dfs直接搜索到底,然后从底部开始向上dp,子树就是子问题,然后在子问题的基础上拓展,求出最优解。

#include <bits/stdc++.h>
#define maxn 300009
#define ll long long

using namespace std;

ll v[maxn];
vector<pair<ll,ll> >node[maxn];
int n;
ll dp[maxn]; 
ll ans = 0;

void dfs(int now,int pre){
	dp[now] = v[now];
	ans = max(dp[now],ans);
	for(int i = 0;i < node[now].size();i++){
		int next = node[now][i].first;
		if(next == pre){
			continue;
		}
		dfs(next,now);
		ans = max(dp[now] + dp[next] - node[now][i].second,ans);
		dp[now] = max(dp[now],dp[next] + v[now] - node[now][i].second);
	}
}

int main(){
	scanf("%d",&n);
	for(int i = 1;i <= n;i++){
		scanf("%lld",&v[i]);
	}
	ll a,b,c;
	for(int i = 0;i < n - 1;i++){
		scanf("%lld %lld %lld",&a,&b,&c);
		node[a].push_back(make_pair(b,c));
		node[b].push_back(make_pair(a,c));
	}
	dfs(1,0);
	printf("%lld\n",ans);
	return 0;
} 

 

没有更多推荐了,返回首页