Codeforces Round #384 (Div. 2) D. Chloe and pleasant prizes 树形DP
D. Chloe and pleasant prizes
链接：
http://codeforces.com/contest/743/problem/D
题解：
对于每个以该点为子树的节点 他都有两种可能的取值 一个是dp[i]，一个为 num[i] 。dp[i] 表示以该点为根的子树当中权值
最大的, num[i]表示以 该点为根的子树 的所有权值（包括自己）;
我们要想使两个子树的和最大,那么就是在以该点为根的子树当中找到两个 和最大的即可。
代码：
1 #include <map> 2 #include <set> 3 #include <cmath> 4 #include <queue> 5 #include <stack> 6 #include <cstdio> 7 #include <string> 8 #include <vector> 9 #include <cstring> 10 #include <sstream> 11 #include <iostream> 12 #include <algorithm> 13 #include <functional> 14 using namespace std; 15 #define rep(i,a,n) for (int i=a;i<=n;i++) 16 #define per(i,a,n) for (int i=n;i>=a;i--) 17 #define pb push_back 18 #define mp make_pair 19 #define all(x) (x).begin(),(x).end() 20 #define SZ(x) ((int)(x).size()) 21 typedef vector<int> VI; 22 typedef long long ll; 23 typedef pair<int, int> PII; 24 const ll mod = 1e9 + 7; 25 //const int inf = 0x3f3f3f3f; 26 const double eps = 1e-7; 27 // head 28 29 const int maxn = 2e5 + 100; 30 vector<int>e[maxn]; 31 ll a[maxn], sum[maxn], dp[maxn], ans; 32 int vis[maxn]; 33 const ll inf = (ll)1e18 + 7; 34 35 void dfs(int rt) { 36 vis[rt] = 1; 37 ll Max = -inf, Min = -inf; 38 sum[rt] = a[rt]; 39 rep(i, 0, int(e[rt].size() - 1)) { 40 int son = e[rt][i]; 41 if (vis[son]) continue; 42 dfs(son); 43 sum[rt] += sum[son]; 44 if (sum[son] > dp[son]) dp[son] = sum[son]; 45 if (dp[son] > dp[rt]) Min = Max, Max = dp[son], dp[rt] = Max; 46 else Min = max(Min, dp[son]); 47 if (Min != -inf) ans = max(ans, Max + Min); 48 } 49 } 50 51 int main() { 52 ios::sync_with_stdio(false); 53 int n, u, v; 54 cin >> n; 55 rep(i, 1, n) cin >> a[i]; 56 rep(i, 1, n) dp[i] = -inf; 57 rep(i, 2, n) { 58 cin >> u >> v; 59 e[u].pb(v); 60 e[v].pb(u); 61 } 62 ans = -inf; 63 dfs(1); 64 if (ans == -inf) cout << "Impossible" << endl; 65 else cout << ans << endl; 66 return 0; 67 }
