算法：树的直径

思路

第一步：从任意一点出发，找到与这一点距离最远的点。
第二步：从最远的一点出发，找到与这点距离最远的点，则这两点之间的距离就是树的直径。

问题描述

很久以前，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≤105,
1≤Pi,Qi≤n,
1≤Di≤1000
输入样例：
5
1 2 2
1 3 1
2 4 5
2 5 4
输出样例：
135

代码

#include <iostream>
#include <cstring>
using namespace std;
const int N = 100010;
int e[N*2], ne[N*2], w[N*2], h[N], idx = 0;
int n;
int dist[N];
void add(int a, int b, int c)
{
    e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx ++ ;
}

void dfs(int u, int father, int distance)
{
    dist[u] = distance;
    for (int i = h[u]; i != -1; i = ne[i])
    {
        int j = e[i];
        if (j != father)
            dfs(j, u, distance + w[i]);
    }
}

int main()
{
    scanf("%d", &n);
    memset(h, -1, sizeof h);
    for (int i = 0; i < n - 1; ++i)
    {
        int a, b, c;
        scanf("%d %d %d", &a, &b, &c);
        add(a, b, c);
        add(b, a, c);
    }
    dfs(1,-1,0);
    int u = 1;
    for (int i = 2; i <= n; ++i)
        if (dist[i] > dist[u])
            u = i;
    dfs(u,-1,0);
    for (int i = 1; i <= n; ++i)
        if (dist[i] > dist[u])
            u = i;
    cout << dist[u] * 10 + dist[u] * (dist[u] + 1ll) / 2 << endl;
    return 0;
}

原题链接

问题描述

给定一个公司的网络，由 n 台交换机和 m 台终端电脑组成，交换机与交换机、交换机与电脑之间使用网络连接。

交换机按层级设置，编号为 1 的交换机为根交换机，层级为 1。

其他的交换机都连接到一台比自己上一层的交换机上，其层级为对应交换机的层级加 1。

所有的终端电脑都直接连接到交换机上。

当信息在电脑、交换机之间传递时，每一步只能通过自己传递到自己所连接的另一台电脑或交换机。

请问，电脑与电脑之间传递消息、或者电脑与交换机之间传递消息、或者交换机与交换机之间传递消息最多需要多少步。

输入格式
输入的第一行包含两个整数 n,m，分别表示交换机的台数和终端电脑的台数。

第二行包含 n−1 个整数，分别表示第 2、3、……、n 台交换机所连接的比自己上一层的交换机的编号。第 i 台交换机所连接的上一层的交换机编号一定比自己的编号小。

第三行包含 m 个整数，分别表示第 1、2、……、m 台终端电脑所连接的交换机的编号。

输出格式
输出一个整数，表示消息传递最多需要的步数。

数据范围
前 30% 的评测用例满足：n≤5,m≤5。
前 50% 的评测用例满足：n≤20,m≤20。
前 70% 的评测用例满足：n≤100,m≤100。
所有评测用例都满足：1≤n≤10000，1≤m≤10000。

输入样例1：
4 2
1 1 3
2 1
输出样例1：
4

代码

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int N = 10010, M = 20010;
int e[M * 2], ne[M * 2], idx = 0, h[M];
int n, m;
int dist[M];
void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}

void dfs(int u, int father, int distance)
{
    dist[u] = distance;
    for (int i = h[u]; i != -1; i = ne[i])
    {
        int j = e[i];
        if (j != father)
            dfs(j, u, distance + 1);
    }
}

int main()
{
    cin >> n >> m;
    memset(h, -1, sizeof h);
    for (int i = 2; i <= n + m; ++i)
    {
        int a;
        cin >> a;
        add(a, i);
        add(i, a);
    }
    dfs(1, -1, 0);
    int u = 1;
    for (int i = 2; i <= n + m; ++i)
        if (dist[u] < dist[i]) u = i;
    dfs(u, -1, 0);
    for (int i = 1; i <= n + m; ++i)
        if (dist[u] < dist[i]) u = i;
        
    cout << dist[u] << endl;
    return 0;
}

原题链接