2019牛客国庆集训派对day4H题-白红宇

2019牛客国庆集训派对day4H题

阅读量：653 次

发布时间：2019-03-15

本文共 1017 字，大约阅读时间需要 3 分钟。

新建一棵树，边权等于原树中(u, v)的唯一路径距离，目标是最大化新树的总成本。

首先，确定原树的直径。直径是指树中延伸最远的简单路径。寻找直径的两个端点可以使用三次DFS：

从任意节点出发，进行DFS，记录每个节点到起点的最远距离。

在第一步中的最远节点作为起点，进行DFS，记录每个节点到该节点的最远距离。

这两个最远节点即为原树的直径端点。

接下来，计算直径端点到其它所有节点的距离之和，即为新树的最大成本。

最佳做法是三次DFS：

#include 
   
    
#include 
    
     
using namespace std;
vector
     
      
       > G[maxn];
vector
       
         visited;
ll dist = -1e18;
int point;
ll res[maxn];
void dfs(int x, bool update, int index) {
    if (update) {
        if (G[x].size() > dist) dist = G[x].size();
        point = x;
    }
    for (int v : G[x]) {
        if (!visited[v]) {
            visited[v] = true;
            if (!update) res[index][v] = res[index][x] + G[x][v];
            else if (dist <= res[index][x] + G[x][v]) {
                res[index][v] = res[index][x] + G[x][v];
                dist = res[index][v];
            }
            dfs(v, update, index);
        }
    }
}