树形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; }