Anniversary party && Computer 树形dp/换根dp

Anniversary party
题解：没什么好讲的,设dp[u][0]代表不选这个节点,dp[u][1]代表选这个节点。

#include<bits/stdc++.h>
using namespace std;
const int maxn = 6005;
int dp[maxn][2],w[maxn];
vector<int>G[maxn];
int dfs(int u,int fa)
{
    dp[u][1] = w[u];
    dp[u][0] = 0;
    for(auto v:G[u])
    if(v!=fa) {
        dfs(v,u);
        dp[u][0] += max(dp[v][1],dp[v][0]);
        dp[u][1] += dp[v][0];
    }
    return max(dp[u][0],dp[u][1]);
}
int main()
{
       int n,u,v;
       while(~scanf("%d",&n))
       {
           for(int i=0;i<maxn;i++)
            G[i].clear();
           for(int i=1;i<=n;i++)
            scanf("%d",&w[i]);
           for(int i=1;i<=n;i++)
           {
               scanf("%d%d",&u,&v);
               if(!u) continue;
               G[u].push_back(v);
               G[v].push_back(u);
           }
           printf("%d\n",dfs(1,0));
       }
}

Computer
题解：换根dp入门
我有一道好题也是换根dp+树直径
题目

#include<bits/stdc++.h>
#define ll long long
#define pi pair<int,int>
#define mk make_pair
using namespace std;
const int maxn = 1e4+10;
vector<pi>G[maxn];
vector<ll>sf[maxn];
ll dp[maxn], ans[maxn];
void dfs(int u,int fa)
{
    dp[u] = 0;
    for(auto tmp:G[u])
    {
        int v = tmp.first , w = tmp.second ;
        if(v==fa) continue;
        dfs(v,u);
        dp[u] = max(dp[u],dp[v]+w);
    }
}
void dfs1(int u,int fa)
{
    sf[u].push_back(0);
    dp[u] = 0;
    for(int i=G[u].size()-1;i>=0;i--)
    {
        int v = G[u][i].first , w = G[u][i].second;
            dp[u] = max(dp[u],dp[v]+w);
            sf[u].push_back(dp[u]);
    }
    ans[u] = dp[u];
    ll x = 0;
    int j = sf[u].size()-2;
    for(auto tmp:G[u])
    {
        int v = tmp.first , w = tmp.second;
        dp[u] = max(x,sf[u][j]);
        x = max(x,dp[v]+w);
        if(v!=fa) dfs1(v,u);
        j--;
    }
}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        int v,w;
        for(int i=1;i<=n;i++)
        G[i].clear(),sf[i].clear();
      for(int i=2;i<=n;i++)
      {
          scanf("%d%d",&v,&w);
          G[i].push_back(mk(v,w));
          G[v].push_back(mk(i,w));
      }
      dfs(1,0);
      dfs1(1,0);
      for(int i=1;i<=n;i++)
        printf("%lld\n",ans[i]);
    }
}