P8602 [蓝桥杯 2013 省 A] 大臣的旅费【树的直径】

P8602 [蓝桥杯 2013 省 A] 大臣的旅费 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

cpp 复制代码
#include<iostream>
#include <algorithm>
#include <vector>
using namespace std;
#define int long long
const int N=5e5+100;
int n;
int res=0;
typedef pair<int,int>pii;
vector<pii>g[N];
bool vis[N];
int len=0;
int dfs(int u)
{
    vis[u]=true;
    int d1=0;//d1记录的是u向下能走的最大长度
    int d2=0;//d2记录的是u向下能走的第二的长度
    for(auto i:g[u])
    {
        int v=i.first;
        int w=i.second;
        if(vis[v])//防止往回搜索
            continue;
        int d=dfs(v)+w;
        if(d>d1)//更新最大值与第二大值
            d2=d1,d1=d;
        else if(d>d2)
            d2=d;
    }
    len=max(len,d1+d2);//保存最大的数值
    return d1;//返回最长长度
}
signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);cout.tie(nullptr);
    cin>>n;
    for(int i=1;i<=n-1;i++)
    {
        int p,q,d;
        cin>>p>>q>>d;
        g[p].push_back((pii){q,d});
        g[q].push_back((pii){p,d});
    }
    dfs(1);//求树的直径,1是根节点
    int ans=0;
    for(int i=1;i<=len;i++)//计算答案
        ans+=10+i;
    cout<<ans;
    return 0;
}
相关推荐
sulikey27 分钟前
C++的STL:深入理解 C++ 的 std::initializer_list
开发语言·c++·stl·list·initializerlist·c++标准库
CoovallyAIHub29 分钟前
超越“识别”:下一代机器视觉如何破解具身智能落地难题?
深度学习·算法·计算机视觉
小白.cpp32 分钟前
list链表容器
数据结构·链表·list
代大大36 分钟前
sciter.js 之cpp使用教程(1)
c++·前端框架
仰泳的熊猫36 分钟前
LeetCode:207. 课程表
数据结构·c++·算法·leetcode
liu****43 分钟前
19.map和set的封装
开发语言·数据结构·c++·算法
孤廖44 分钟前
C++ 模板再升级:非类型参数、特化技巧(含全特化与偏特化)、分离编译破解
linux·服务器·开发语言·c++·人工智能·后端·深度学习
水冗水孚44 分钟前
双指针算法在实际开发中的具体应用之代码Review文章字符串的片段分割
算法·leetcode
DuHz44 分钟前
用于汽车雷达应用的步进频率PMCW波形——论文阅读
论文阅读·算法·汽车·信息与通信·信号处理·毫米波雷达
张晓~183399481211 小时前
碰一碰发抖音源码技术搭建部署方案
线性代数·算法·microsoft·矩阵·html5