【蓝桥杯】蓝桥杯双周赛第二场E题

知识点:树的直径

题目

过年了。

蓝桥村可以抽象为n个节点,n - 1条边的一棵树,每条边有边权长度wi。

小蓝可以选择任意一个点作为起点,然后选择一条路径,可以访问每一个节点最少一次。他想知道最短的路径长度是多少。

输入格式

第一行输入一个整数n,表示节点的数量。

加下来n - 1行,每行三个整数vi,ui,wi,表示(vi,ui)存在一条wi的边。

输出格式

输出一个整数,表示最短路径。

思路

我们可以从任意一个节点开始,必须至少到达每个节点最少一次。我们很容易得到最短路程是所有道路长度之和的两倍减去这棵树的直径。数的直径可以从树中任意一点出发,找到距离这个点L(第一遍DFS),再从L出发到达树中距离S最远的点L,S到L这段距离就是树的直径。

代码

cpp 复制代码
#include<bits/stdc++.h>
#define int long long
#define N 200010 // 要建无向边,开两倍大小
using namespace std;

int n;
int e[N],ne[N],h[N],w[N],idx;// 邻接表四件套
int dist[N];// 该点到起始点的距离
bool st[N];// 判断该点是否被便利过
int ans;
void add(int a,int b,int c)
{
    w[idx] = c,e[idx] = b,ne[idx] = h[a],h[a] = idx ++;
}

void dfs(int u)
{
    st[u] = true;
    for(int i = h[u]; i != -1; i = ne[i])
    {
        int j = e[i];
        if(!st[j] && dist[j] > dist[u] + w[i])
        {
            dist[j] = dist[u] + w[i];
            st[j] = true;
            dfs(j);
        }
    }
}

int32_t main()
{
    memset(h,-1,sizeof(h));
    cin >> n;
    for(int i = 0; i < n - 1; i ++)// 建树,无向边
    {
        int a,b,c;
        cin >> a >> b >> c;
        add(a,b,c);
        add(b,a,c);
        ans += c * 2;
    }
    memset(dist,0x3f,sizeof(dist));
    dist[1] = 0;// 从任意一点开始
    dfs(1);
    int maxx = -1,address = -1;
    for(int i = 1; i <= n; i ++)// 找到距离这个点最远距离的点
    {
        if(dist[i] > maxx)
        {
            maxx = dist[i];
            address = i;
        }
    }
    for(bool & i : st)// 将数据重置
    {
        i = false;
    }
    memset(dist,0x3f,sizeof(dist));
    dist[address] = 0;
    dfs(address);
    maxx = -1;
    for(int i = 1; i <= n; i ++)// 找到最远的距离(树的直径)
    {
        if(dist[i] > maxx)
        {
            maxx = dist[i];
        }
    }
    cout << ans - maxx << endl;// 所有路程之和的两倍减去树的直径就是最短距离
    return 0;
}
相关推荐
向阳@向远方15 分钟前
第二章 简单程序设计
开发语言·c++·算法
Mr_Xuhhh1 小时前
信号与槽的总结
java·开发语言·数据库·c++·qt·系统架构
github_czy1 小时前
RRF (Reciprocal Rank Fusion) 排序算法详解
算法·排序算法
liulilittle1 小时前
VGW 虚拟网关用户手册 (PPP PRIVATE NETWORK 基础设施)
开发语言·网络·c++·网关·智能路由器·路由器·通信
许愿与你永世安宁2 小时前
力扣343 整数拆分
数据结构·算法·leetcode
爱coding的橙子2 小时前
每日算法刷题Day42 7.5:leetcode前缀和3道题,用时2h
算法·leetcode·职场和发展
ruanjiananquan992 小时前
c,c++语言的栈内存、堆内存及任意读写内存
java·c语言·c++
满分观察网友z2 小时前
从一次手滑,我洞悉了用户输入的所有可能性(3330. 找到初始输入字符串 I)
算法
持梦远方2 小时前
C 语言基础入门:基本数据类型与运算符详解
c语言·开发语言·c++
YuTaoShao3 小时前
【LeetCode 热题 100】73. 矩阵置零——(解法二)空间复杂度 O(1)
java·算法·leetcode·矩阵