B - 树上子链 （带权树的直径，树形DP）

传送门

---

题目大意

给出一棵结点带权的树，定义树上的最长子链为子链上权值和最大的子链，求出最长子链的权值和。

解题思路

最近在搞带权树的直径，搜到了这道题，就写一下吧

首先不难发现是树形DP。这道题的状态应该是很好定义的，因为权值在点上而不是边上。首先考虑的一点应该是将这棵树变为有根树，那么设$d(u)$表示以节点$u$为根，延伸到叶子节点的最长路径的长度，因为题目说路径可以是节点本身，那么初始化就是这个点的权值。然后考虑状态转移方程，显然是考虑哪个叶子节点到该根节点的长度最长，那么就是$d(u)=max(d(u),d(v)+w[u])$

写到这里本题似乎迎刃而解，但是有一个需要仔细考虑的细节：在哪里更新答案？首先知道如果叶子节点就是起点，该节点就是终点，那么只需在每行代码的最后更新一下$ans=max(ans,d(u))$，但是也有可能是$u$节点为中间节点，那么最长的路径应该考虑该根节点下的两个叶子连成的路径，即$ans=max(ans,d(u)+d(v))$

然后更新的顺序需要注意，见代码注释

//
// Created by Happig on 2020/8/19
//
#include <bits/stdc++.h>
#include <unordered_map>
#include <unordered_set>

using namespace std;
#define fi first
#define se second
#define pb push_back
#define ins insert
#define Vector Point
#define lowbit(x) (x&(-x))
#define mkp(x, y) make_pair(x,y)
#define mem(a, x) memset(a,x,sizeof a);
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef pair<double, double> pdd;
const double eps = 1e-8;
const double pi = acos(-1.0);
const int inf = 0x3f3f3f3f;
const double dinf = 1e300;
const ll INF = 1e18;
const int Mod = 1e9 + 7;
const int maxn = 2e5 + 10;

vector<int> G[maxn];
ll d[maxn], w[maxn];
ll ans;

void dfs(int u, int fa) {
    d[u] = w[u];
    for (auto v: G[u])
        if (v != fa) {
            dfs(v, u);
            ans = max(ans, d[u] + d[v]);  //应该先更新答案再更新d(u)，否则结果会变大错误
            d[u] = max(d[u], d[v] + w[u]);
        }
    ans = max(ans, d[u]);
}

int main() {
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int n;
    cin >> n;
    for (int i = 1; i <= n; i++) cin >> w[i];
    for (int i = 1, u, v; i < n; i++) {
        cin >> u >> v;
        G[u].push_back(v);
        G[v].push_back(u);
    }
    ans = -inf;
    dfs(1, -1);
    cout << ans << "\n";
    return 0;
}