HDU-6201 transaction transaction transaction(树形dp)

最新推荐文章于 2021-01-28 08:29:02 发布

薄层

最新推荐文章于 2021-01-28 08:29:02 发布

阅读量309

点赞数 1

分类专栏：树形DP HDU 文章标签：树形dp

本文链接：https://blog.csdn.net/yo_bc/article/details/77926290

版权

HDU 同时被 2 个专栏收录

92 篇文章 0 订阅

订阅专栏

树形DP

2 篇文章 0 订阅

订阅专栏

2017 ACM/ICPC Asia Regional Shenyang Online

题意：

给定一个树形图，每个节点都有一个商品价格，走每条边都有一定的花费，问商人任选一个节点购买商品然后去另一个节点卖商品所能获得的最大利润。

思路：

树形dp。任选一个节点作为根节点进行dfs，对于每个节点记录其两个状态：dp[u][0]代表选取该节点及其子孙中任一节点作为起点到达当前节点所要花费的最小代价，dp[u][1]代表选取该节点及其子孙中任一节点作为终点到达当前节点所能获得的最大收入，对于每个节点，通过dp[u][1]-dp[u][0]更新答案取得最优。

代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
struct node{int v, w;};
vector<node> G[maxn];
int dp[maxn][2]; 
int t, n, ans;
int val[maxn];
void dfs(int u, int fa)
{
	dp[u][0] = dp[u][1] = val[u];
	for(int i = 0; i < G[u].size(); ++i)
	{
		int v = G[u][i].v;
		if(v == fa) continue;
		dfs(v, u);
		dp[u][0] = min(dp[u][0], dp[v][0]+G[u][i].w);
		dp[u][1] = max(dp[u][1], dp[v][1]-G[u][i].w);
	}
	ans = max(ans, dp[u][1]-dp[u][0]);
}
int main()
{
	for(scanf("%d", &t); t--;)
	{
		scanf("%d", &n); ans = 0;
		for(int i = 1; i <= n; ++i)
		scanf("%d", &val[i]), G[i].clear();
		for(int i = 1; i < n; ++i)
		{
			int u, v, w;
			scanf("%d %d %d", &u, &v, &w);
			G[u].push_back((node){v, w});
			G[v].push_back((node){u, w});
		}
		dfs(1, -1);
		printf("%d\n", ans);
	}
	return 0;
}

继续加油~