SGU - 143 Long Live the Queen 树形dp之加和问题

树形dp加和问题；

从一个点开始dfs，以这个点为根，对他的子节点dfs，他的子节点遍历的时候不会包含他本身，等下证明正确性；如果他的子节点作为根遍历完了以后，子节点的值大于0时，将这个值加到当前结点；

证明正确性：首先如果这个点权值是负数，那对他子节点dfs的时候，没必要加上他；如果这个点的权值是正数，算完后可以的话直接加在上面就行了；

#include<bits/stdc++.h>
using namespace std;
const int maxn = 16000 + 7, maxd = 3000 + 7;;
typedef long long ll;
const int INF = 0x3f3f3f3f;

int n;
int a[maxn];
vector<int> vec[maxn];
int dp[maxn];


void dfs(int u, int f) {
    for(auto i : vec[u]) {
        if(i == f) continue;
        dfs(i, u);
        dp[u] += (dp[i] > 0 ? dp[i] : 0);
    }
    return;
}

int main() {
    scanf("%d", &n);
    for(int i = 1; i <= n; ++i) { scanf("%d", &a[i]); dp[i] = a[i];}
    for(int i = 1; i < n; ++i) {
        int u, v; scanf("%d%d", &u, &v);
        vec[u].push_back(v);
        vec[v].push_back(u);
    }
    dfs(1,0);
    int ans = -INF;
    for(int i = 1; i <= n; ++i) {
        ans = max(ans, dp[i]);
    }
    printf("%d", ans);
    return 0;
}