树的直径 AcWing 1207. 大臣的旅费

题目描述

很久以前，T王国空前繁荣。

为了更好地管理国家，王国修建了大量的快速路，用于连接首都和王国内的各大城市。

为节省经费，T国的大臣们经过思考，制定了一套优秀的修建方案，使得任何一个大城市都能从首都直接或者通过其他大城市间接到达。

同时，如果不重复经过大城市，从首都到达每个大城市的方案都是唯一的。

J是T国重要大臣，他巡查于各大城市之间，体察民情。

所以，从一个城市马不停蹄地到另一个城市成了J最常做的事情。

他有一个钱袋，用于存放往来城市间的路费。

聪明的J发现，如果不在某个城市停下来修整，在连续行进过程中，他所花的路费与他已走过的距离有关，在走第x千米到第x+1千米这一千米中（x是整数），他花费的路费是x+10这么多。也就是说走1千米花费11，走2千米要花费23。

J大臣想知道：他从某一个城市出发，中间不休息，到达另一个城市，所有可能花费的路费中最多是多少呢？

输入格式

输入的第一行包含一个整数 n，表示包括首都在内的T王国的城市数。

城市从 1 开始依次编号，1 号城市为首都。

接下来 n−1 行，描述T国的高速路（T国的高速路一定是 n−1 条）。

每行三个整数 Pi,Qi,Di，表示城市 Pi 和城市 Qi 之间有一条双向高速路，长度为 Di 千米。

输出格式

输出一个整数，表示大臣J最多花费的路费是多少。

数据范围

1≤n≤10^5,
1≤Pi,Qi≤n,
1≤Di≤1000

输入样例：

5 
1  2  2 
1  3  1 
2  4  5 
2  5  4 

输出样例：

135

分析题目

下面的方法与证明都在树的各条边权值大于0的情况的前提下才成立！！！

 题意大概就是在一个树中，求出一条权值总和最大的路径，也就是树的直径，这也就是树的直径的定义

如何求一个树的直径呢，首先给出结论，我们以任意一点开始，找到离这个点最远的点，然后用找到的这个点，以找到的这个点为起点再进行上述操作，得到的一条路径就是树的直径。

证明

我们用反证法来证明，假设以x点为起点找到离x点最远的点y，然后以y点开始找到离y点最远的点不是直径，直径为一条点a到点b的路径。

第一种情况是直径与x点到y点的路径有交点p

如下图所示

 因为点y是我们找到的离点x最远的一个点，所以xp+py>xp+pb,也就有py>pb,那么ap+py>ap+pb，与我们的假设矛盾，我们能找到一条更长的边比ab要长，所以可以得到，这种情况时，以y开始找到的点一定是树的直径，也就是权值总和最大的路径

第二种情况直径与xy没有交点，因为这是一颗树，那么一定存在一条边pc连接ab和xy，如下图所示

 y是我们找到离x最远的一条边，那么xp+py>xp+pc+cb,即py>pc+cb,所以一定有py>cb，则ac+cp+py>ac+cb，与我们的假设矛盾，我们同样能找到一条更长的边比ab要长，因此可以得出我们的结论一定是正确的

代码实现

上面证明完了，接下来是讲如何实现，我们以任意一点，用dfs找到这点能到达的点的所有权值，然后找到权值最大的那个点，也就是离这个点最远的那个点，接着再以找到的那个点重复一遍上述操作就可以了 

代码如下：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;

const int N=1e5+10;

int n;
int h[N],e[N*2],ne[N*2],w[N*2],idx;//一定要注意，题目给的是无向边，所以数组要开两倍大小
int dist[N];

void add(int a,int b,int x)//邻接表存储树
{
    e[idx]=b;
    ne[idx]=h[a];
    w[idx]=x;//w存储边的权值
    h[a]=idx++;
}

//三个参数分别代表此刻的节点u，节点u的父节点，从起始节点到节点u的长度
void dfs(int u,int father,int distance)
{
    dist[u]=distance;//给节点u赋值，表示离起始点的距离
    for(int i=h[u];i!=-1;i=ne[i])//寻找u的子节点，更新距离
    {
        int j=e[i];//u的子节点
        if(j!=father)//要保证往后找，不能往回找，也就是找过的点不能再找了
        {
            dfs(j,u,distance+w[i]);
        }
    }
}
int main()
{
    memset(h,-1,sizeof h);//邻接表头初始化
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        int u,v,x;
        scanf("%d%d%d",&u,&v,&x);
        add(u,v,x);//无向边，要add两次
        add(v,u,x);
    }
    dfs(1,-1,0);//以任意点去找离他最远的点，这里我们以1为起点，-1表示1没有父节点
    int u=1;
    for(int i=1;i<=n;i++)//找到离1号点最远的点u
        if(dist[i]>dist[u])
            u=i;
    dfs(u,-1,0);//找到离u点最远的点
    u=1;
    for(int i=1;i<=n;i++)//找到离u点最远的点
        if(dist[i]>dist[u])
            u=i;
    int s=dist[u];//s表示树的直径
    //这里与题意有关，这里有个小技巧，乘积可能会爆int，我们给1后面加个ll就可以将这个数自动转换成long long
    printf("%lld",10*s+(s+1ll)*s/2);
    return 0;
}