P1122 最大子樹和——（樹形DP）

總結

第一次寫樹形DP，找了一個簡單的題，先熟悉一下套路，雖然自己的這種寫法，過了，可以求到最大的子樹和，但是不能求到每個結點的最大子樹和，只有某一部分的最大子樹和的根，它是正確的，其他結點都是打工仔。

解析

遞歸下去，以子樹推當前結點和子樹的最優值，是可以辦到的
情況1：如果當前子樹<=0，就不更新
情況2：如果當前子樹>0,就更新上去。

題目鏈接

//#pragma GCC optimize(2)
#include<bits/stdc++.h>
//typedef long long ll;
//#define ull       unsigned long long
#define int       long long
#define F           first
#define S           second
#define endl        "\n"//<<flush
#define eps         1e-6
#define lowbit(x)   (x&(-x))
#define PI          acos(-1.0)
#define inf         0x3f3f3f3f
#define MAXN        0x7fffffff
#define INF         0x3f3f3f3f3f3f3f3f
#define pa          pair<int,int>
#define ferma(a,b)  pow(a,b-2)
#define pb          push_back
#define all(x)      x.begin(),x.end()
#define memset(a,b) memset(a,b,sizeof(a));
#define IOS         ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;
void file()
{
#ifdef ONLINE_JUDGE
#else
    freopen("cin.txt","r",stdin);
    //  freopen("cout.txt","w",stdout);
#endif
}
const int N=2e4+5;
vector<int>G[N];
int a[N],dp[N],ans;
void dfs(int u,int fat)
{
    int len=G[u].size();
    dp[u]=a[u];
    for(int i=0;i<len;++i)
    {
        int v=G[u][i];
        if(v!=fat)
        {
            dfs(v,u);
            if(dp[v]>=0)
                dp[u]+=dp[v];
        }
    }
    ans=max(ans,dp[u]);
}
signed main()
{
    IOS;
    //file();
    int n;
    cin>>n;
    for(int i=1;i<=n;++i)
        cin>>a[i],ans+=a[i];
    for(int i=1;i<n;++i)
    {
        int x,y;
        cin>>x>>y;
        G[x].pb(y);
        G[y].pb(x);
    }
    dfs(1,0);
    cout<<ans<<endl;
    return 0;
}