[不知道出自什么OJ]最大利润

题目描述

政府邀请了你在火车站开饭店，但不允许同时在两个相连接的火车站开。任意两个火车站有且只有一条路径，每个火车站最多有50个和它相连接的火车站。

告诉你每个火车站的利润，问你可以获得的最大利润为多少。

输入格式

第一行输入整数N（N≤100000），表示有N个火车站，分别用1，2，... ，N来编号。

接下来N行，每行一个整数（不超过10000）表示每个站点的利润。

接下来N-1行描述火车站网络，每行两个整数，表示相连接的两个站点。

输出格式

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

题目分析：

树形\(dp\)入门题 为了讲课码的

设\(dp[i][0/1]\)表示第\(i\)个节点选/不选的最优解

方程显然，懒了，见代码

代码：

#include<bits/stdc++.h>
#define N (300000 + 10)
using namespace std;
inline void read(int &cnt) {
	cnt = 0;
	int f = 1; char c = getchar();
	while (!isdigit(c)) {if (c == '-') f = -f; c = getchar();}
	while (isdigit(c)) {cnt = (cnt << 3) + (cnt << 1) + (c ^ 48); c = getchar();}
	cnt *= f;
}
int n, first[N], to[N], nxt[N], tot, a, b;
int dp[N][2];
void add(int x, int y) {nxt[++tot] = first[x], first[x] = tot, to[tot] = y;}

void dfs_(int p, int fa) {
	for (register int i = first[p]; i; i = nxt[i]) {
		int v = to[i];
		if (v == fa) continue;
		dfs_(v, p);
		dp[p][0] += max(dp[v][0], dp[v][1]);
		dp[p][1] += dp[v][0];
	}

}

int main() {
	read(n);
	for (register int i = 1; i <= n; ++i) read(dp[i][1]);
	for (register int i = 1; i < n; ++i) read(a), read(b), add(a, b), add(b, a);
	dfs_(1, 0);
	printf("%d", max(dp[1][0], dp[1][1]));
	return 0;
}