题面
【题目描述】
政府邀请了你在火车站开饭店,但不允许同时在两个相连的火车站开。任意两个火车站有且只有一条路径,每个火车站最多有 50 个和它相连接的火车站。
告诉你每个火车站的利润,问你可以获得的最大利润为多少?
例如下图是火车站网络:
最佳投资方案是 1 , 2 , 5 , 6 这 4 个火车站开饭店可以获得的利润为 90.
【输入格式】
第一行输入整数 N(<=100000), 表示有 N 个火车站,分别用 1,2,……..,N 来编号。接下来 N 行,每行一个整数表示每个站点的利润,接下来 N-1 行描述火车站网络,每行两个整数,表示相连接的两个站点。
【输出格式】
输出一个整数表示可以获得的最大利润。
【样例输入】
6
10
20
25
40
30
30
4 5
4 6
3 4
1 3
2 3
【样例输出】
90
- 由题意知这是一棵树
- 不允许在相邻的火车站开店的意思是任意两个店不能直接相连
- 用f[i]表示选这个点所获得的最大利润,g[i]表示不选这个点所获得的最大利润
- DP方程:
f[u]+=g[v]; 不选子节点
g[u]+=max(g[v],f[v]); 比较利润确定选不选子节点
f[u]+=val[u]; 若选这个点,则加上这个店的利润
这个代码也可以过cogs 190 周年纪念聚会
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=100010;
int n;
struct node{
int to,next;
}e[MAXN*2];
int head[MAXN],cnt=0;
int val[MAXN];
int f[MAXN],g[MAXN];
void add(int f,int t){//邻接表存边
cnt++;
e[cnt]=(node){t,head[f]};
head[f]=cnt;
}
void dfs(int u,int fa){
for(int i=head[u];i;i=e[i].next){
int v=e[i].to;
if(v!=fa){//判重
dfs(v,u);
f[u]+=g[v];
g[u]+=max(g[v],f[v]);
}
}
f[u]+=val[u];
}
int main(){
freopen("profitz.in","r",stdin);
freopen("profitz.out","w",stdout);
scanf("%d",&n);
for(int i=1;i<=n;++i) scanf("%d",&val[i]);
for(int i=1;i<=n-1;++i){
int a,b;
scanf("%d%d",&a,&b);
add(a,b);add(b,a);
}
dfs(1,1);
cout<<max(f[1],g[1]);
return 0;
}