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]表示以 该点为根的子树 的所有权值（包括自己）;

我们要想使两个子树的和最大,那么就是在以该点为根的子树当中找到两个 和最大的即可。

代码：

#include <map>
#include <set>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <string>
#include <vector>
#include <cstring>
#include <sstream>
#include <iostream>
#include <algorithm>
#include <functional>
using namespace std;
#define rep(i,a,n) for (int i=a;i<=n;i++)
#define per(i,a,n) for (int i=n;i>=a;i--)
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define SZ(x) ((int)(x).size())
typedef vector<int> VI;
typedef long long ll;
typedef pair<int, int> PII;
const ll mod = 1e9 + 7;
//const int inf = 0x3f3f3f3f;
const double eps = 1e-7;
// head

const int maxn = 2e5 + 100;
vector<int>e[maxn];
ll a[maxn], sum[maxn], dp[maxn], ans;
int vis[maxn];
const ll inf = (ll)1e18 + 7;

void dfs(int rt) {
    vis[rt] = 1;
    ll Max = -inf, Min = -inf;
    sum[rt] = a[rt];
    rep(i, 0, int(e[rt].size() - 1)) {
        int son = e[rt][i];
        if (vis[son]) continue;        
        dfs(son);
        sum[rt] += sum[son];
        if (sum[son] > dp[son]) dp[son] = sum[son];
        if (dp[son] > dp[rt]) Min = Max, Max = dp[son], dp[rt] = Max;
        else Min = max(Min, dp[son]);
        if (Min != -inf) ans = max(ans, Max + Min);
    }
}

int main() {
    ios::sync_with_stdio(false);
    int n, u, v;
    cin >> n;
    rep(i, 1, n) cin >> a[i];
    rep(i, 1, n) dp[i] = -inf;
    rep(i, 2, n) {
        cin >> u >> v;
        e[u].pb(v);
        e[v].pb(u);
    }
    ans = -inf;
    dfs(1);
    if (ans == -inf) cout << "Impossible" << endl;
    else cout << ans << endl;
    return 0;
}