DFS(树的直径)-第四届蓝桥杯省赛C++A组-大臣的旅费

DFS(树的直径)-第四届蓝桥杯省赛C++A组-大臣的旅费

题目：

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

题意：

$n个节点，任何一个节点P_i都可以通过1号节点到达节点Q_i，长度为D_i，花费为{\sum }_{j=1}^{Di}(10+j)=10D_i+\frac{(D_i+1)D_i}{2}。$

$问所有的通路当中，花费最大的时多少，也就是最长的一条路是多长。$

$整个图可以看作一个以1号节点为根的树，要求树的直径。$

$任选一个点u，dfs到u的最远的点v，再从点v开始dfs到v的最远距离即直径。就是找树上距离最远的两个边界点。$

$建图：$

$①、vector。$

$②、数组模拟邻接表，即n个单链表$——《单链表的数组实现》。

---

代码：

①、vector：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<vector>
#define P pair<int,int>
#define x first
#define y second
#define ll long long
#define inf 0x7fffffff
using namespace std;
const int N=1e5+10;
int n;
struct Edge
{
    int id,w;
};
vector<Edge>V[N];
int dis[N];

void dfs(int u,int father,int distance)
{
    dis[u]=distance;

    for(auto e :V[u])
        if(e.id!=father)
            dfs(e.id,u,distance+e.w);
}

int main()
{
    int n;
    cin>>n;
    for(int i=0;i<n-1;i++)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        V[a].push_back({b,c});
        V[b].push_back({a,c});
    }

    dfs(1,-1,0);
    ///找到到点1距离最大的点
    int max_pos=1;
    for(int i=1;i<=n;i++)
        if(dis[i]>dis[max_pos])
            max_pos=i;

    ///找到到点max_pos距离最大的点
    dfs(max_pos,-1,0);
    for(int i=1;i<=n;i++)
        if(dis[i]>dis[max_pos])
            max_pos=i;

    int s=dis[max_pos];

    printf("%lld\n",s*10+s*(s+1ll)/2);

    return 0;
}

---

②、数组模拟：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 100010, M = 200010;

int n;
int h[N], e[M], w[M], ne[M], idx;  ///h邻接表头节点，e终点，w权值，ne下一个相连终点，idx当前插入节点的编号
int dist[N];

void add(int a, int b, int c)
{
    e[idx] = b;
    w[idx] = c;
    ne[idx] = h[a]; 
    h[a] = idx;
    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);   ///所有头节点置-1

    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[u] < dist[i])
            u = i;

    dfs(u, -1, 0);

    for (int i = 1; i <= n; i ++ )
        if (dist[u] < dist[i])
            u = i;

    printf("%lld\n", dist[u] * 10 + (dist[u] + 1ll) * dist[u] / 2);

    return 0;
}