传送门:Computer
题意:有 n 个点,通过 n-1 条边连成一棵树,每条边有一个权值,求从每个点出发的简单路径的最大长度。
思路:树形dp
首先以 1 为根建树,这样每个点能走的最大路径有两种情况:往下沿着子节点走或往上沿着父结点走。假设根节点为 rt,我们定义:
dp[rt][0] 表示以 rt 为根的子树上的节点到 rt 的最长距离。
dp[rt][1] 表示以 rt 为根的子树上的节点到 rt 的次长距离。
dp[rt][2] 表示从 rt 往上走能走的最长距离。
我们先通过一次 dfs 求出 dp[rt][0] 与 dp[rt][1]。此时只需要简单的 dfs 和更新就可以了。
然后我们通过第二次 dfs 求出 dp[rt][2]。我们假设 to 是 rt 的一个子节点,此时我们需要分两种情况考虑:
dp[rt][0] 经过了节点 to,那么从 to 往上走的最长的路可能是【连接 rt 和 to 的边】加上(【rt 往上走的最长的路】或【rt 往下走的次长的路】)。
dp[rt][0] 没有经过节点 to,那么从 to 往上走的最长的路可能是【连接 rt 和 to 的边】加上(【rt 往上走的最长的路】或【rt 往下走的最长的路】)。
即:
dp[to][2]=e[i].w+max(dp[rt][2],dp[rt][1])(dp[rt][0] 经过节点 to)
dp[to][2]=e[i].w+max(dp[rt][2],dp[rt][0])(dp[