1487. 最大利润

最新推荐文章于 2023-09-07 13:11:15 发布

原创最新推荐文章于 2023-09-07 13:11:15 发布 · 158 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#树形DP

信息学专栏收录该内容

77 篇文章

订阅专栏

博客围绕火车站开饭店问题展开，政府规定不能在相连火车站同时开店，每个站有对应利润。给出火车站数量、各站利润及连接关系，要求计算可获最大利润，还给出了输入输出格式。

题目描述

政府邀请了你在火车站开饭店，但不允许同时在两个相连接的火车站开。任意两个火车站有且只有一条路径，每个火车站最多有50个和它相连接的火车站。
告诉你每个火车站的利润，问你可以获得的最大利润为多少。
例如下图是火车站网络：
在这里插入图片描述
最佳投资方案是在1，2，5，6这4个火车站开饭店可以获得利润为90

输入

第一行输入整数N(<=100000)，表示有N个火车站，分别用1，2。。。，N来编号。接下来N行，每行一个整数表示每个站点的利润，接下来N-1行描述火车站网络，每行两个整数，表示相连接的两个站点。

输出

输出一个整数表示可以获得的最大利润。

Code

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
int point[6002010],x,y,tot = 0;
struct node
{
	int to,next;
}a[200010];
long long stu[600010],f[600010][2];
inline void dp(int cur,int fa)
{
	f[cur][1] = point[cur]; 
	f[cur][0] = 0;
	for(int i = stu[cur];i != -1;i = a[i].next)
	{
		if(fa == a[i].to) 
			continue;
		dp(a[i].to,cur);
		f[cur][1] += f[a[i].to][0];
		f[cur][0] += max(f[a[i].to][0],f[a[i].to][1]);
	}
}
int main()
{
	int n;
	freopen("profit.in","r",stdin);
	freopen("profit.out","w",stdout);
	memset(stu,-1,sizeof stu);
	scanf("%d",&n);
	for(int i = 1;i <= n;++i)
		scanf("%d",&point[i]);
	for(int i = 1;i < n;++i)
	{
		scanf("%d%d",&x,&y);
		a[++tot].to = y; 
		a[tot].next = stu[x]; 
		stu[x] = tot;
		a[++tot].to = x; 
		a[tot].next = stu[y];
		stu[y] = tot;
	}
	dp(1,0);
	printf("%d",max(f[1][1],f[1][0]));
	return 0;
}