洛谷P1122最大子树和题解

题目

一道比较好想的树形\(DP\)

完全可以用树形DP的基本思路，递归，然后取最优的方法。

\(Code\)

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
using namespace std;
int n, a, b, cnt, maxn, data[100100], dp[100100], lin[100010];
struct cym {
    int to, nex;
}e[100100];
inline void add(int f, int t)
{
    e[++cnt].to = t;
    e[cnt].nex = lin[f];
    lin[f] = cnt;
}
inline void dfs(int now, int fa)
{   
    dp[now] = data[now];
    for (int i = lin[now]; i; i = e[i].nex)
    {
        int to = e[i].to;
        if (to == fa) continue;
        dfs(to, now);
        dp[now] += max(0, dp[to]);
    }
    maxn = max(maxn, dp[now]);
}
int main()
{
    scanf("%d", &n);
    for (int i = 1; i <= n; i++)
        scanf("%d", &data[i]), dp[i] = data[i];
    for (int i = 1; i < n; i++) 
    {
        scanf("%d%d", &a, &b);
        add(a, b), add(b, a);
    }
    dfs(1, 0);
/*  for (int i = 1; i <= n; i++)
        if (!dp[i])
            dfs(i, 0);
*/
    printf("%d", maxn); 
}