牛客 树上子链 dp求树的直径

牛客 树上子链 链接

题意

给出一个树，n点，n-1条边，树上每个点有权重，求这个树权重之和最大的链。就是求树的直径，不过有负数。

输入

输出

仅包含一个数，表示我们所需要的答案。

样例

题解

树的直径dp版的模板
因为两遍bfs求树的直径必须保证所有的权值为正，对于既有正数又有负数的情况，无法求得，因为从某一点出发，尽可能延伸到最大，当出现负数时就会限制延伸的区域，从而求得的结果只是局部最优解。
正题：如何树形dp
树形dp关键在于如何找到，子树的转移方程以及dp需要转移的状态。首先，从一颗子树思考，当前子树的直径是多少。

子树A的直径为BCD中最大的两个Max len 之和加上A的权值，Max len B定义为从B点出发，到各个子节点的最长距离。其他点类似。
对于这样一颗简单的树，这样求没什么问题，但是。。

假设每个点的权值为1，那么对于A子树，B和C的Maxlen最大，因此A子树其直径为MaxlenB+MaxlenC+WeightA=2+2+1=5，同理对于O子树其直径为MaxlenA+WeightO=3+1=4，那么从O点出发得到的结果是4，显然是不对的，而答案是5。因此修正一下，子树直径算法不应该是，只选择当前子树，前两条最大的链，还需要考虑其所有子树的直径，然后取最大值，和点分治dp过程是一致的。如果要得到整个树的直径，需要对每个“子树直径”取一个最大值。
设dp[n][0]记录“子树直径”，dp[n][1]记录Maxlen
dp[cur][0]=weight[cur]+第一大dp[to][1]+第二大dp[to][1]
dp[cur][1]=max{dp[to][1]}
ans = max{dp[i->n][0]}

代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define maxn 100005
#define INF 214748364700LL
#define IOS ios::sync_with_stdio(false)

vector<ll>V[maxn];
ll w[maxn],dp[maxn][2],ans;
void dfs(ll cur,ll fa)
{
    ll Max1=-INF,Max2=-INF;  // 记录最大值和次大值
    for(ll i=0; i<V[cur].size(); i++)
    {
        ll to=V[cur][i];
        if(to!=fa)
        {
            dfs(to,cur);
            
            if(Max1<=dp[to][1])
            {
                if(Max1!=-INF)
                    Max2=Max1;
                Max1=dp[to][1];
            }
            else
                Max2=max(Max2,dp[to][1]);
        }
    }
    
    //因为存在负数，Max只考虑大于0
    if(Max1>0)
        dp[cur][1]=w[cur]+Max1;
    else
        dp[cur][1]=w[cur];

    dp[cur][0]=w[cur];
    if(Max1>0)
        dp[cur][0]+=Max1;
    if(Max2>0)
        dp[cur][0]+=Max2;
    ans=max(ans,dp[cur][0]);
}


int main()
{
    ll n;
    cin>>n;

    for(ll i=1; i<=n; i++)
        cin>>w[i];

    for(ll i=1; i<=n-1; i++)
    {
        ll x,y;
        cin>>x>>y;
        V[x].push_back(y);
        V[y].push_back(x);
    }

    ans=-INF;
    dfs(1,1);
    cout<<ans;
    return 0;
}