小A与欧拉路(树直径)

小A与欧拉路

https://ac.nowcoder.com/acm/problem/22618

题目:

小A给你了一棵树,对于这棵树上的每一条边,你都可以将它复制任意(可以为0)次(即在这条边连接的两个点之间再加一条边权相同的边),求所有可能新形成的图中欧拉路的最短长度。
欧拉路:从图中任意一个点开始到图中任意一个点结束的路径,并且图中每条边只通过恰好一次。



做法:

不难发现每条树边顶多用次,而且路径的起点和终点都是叶子。假如我们确定个叶子作为欧拉路的起终点。那么路径上的边算次,其他边算次。所以我们让这个路径最长,也就是选取树的某条直径端点作为欧拉路的两端,在直径上的边算次贡献,不在则次,就是答案。


代码:

#include <bits/stdc++.h>
#define IOS ios::sync_with_stdio(false), cin.tie(0)
#define debug(a) cout << #a ": " << a << endl
using namespace std;
typedef long long ll;
const int N = 2e5 + 7;
vector<pair<int,int> > g[N];
int rt1, rt2, dis[N], fa[N], mrk[N];
void dfs(int u, int p, int& rt){
    for (auto &pr : g[u]) if (pr.first != p){
        int v = pr.first, w = pr.second;
        fa[v] = u, mrk[v] = w;
        dis[v] = dis[u] + w;
        if (dis[v] > dis[rt]) rt = v;
        dfs(pr.first, u, rt);
    }
}
int main(void){
    IOS;
    int n; cin >> n;
    ll ans = 0;
    for (int i = 0; i < n-1; ++i){
        int u, v, w; cin >> u >> v >> w;
        g[u].push_back(make_pair(v, w));
        g[v].push_back(make_pair(u, w));
        ans += 2*w;
    }
    dfs(1, 0, rt1); dis[rt1] = 0; dfs(rt1, 0, rt2);
    //debug(rt1), debug(rt2);
    while (rt2 != rt1){
        ans -= mrk[rt2], rt2 = fa[rt2];
    }
    cout << ans << endl;
    return 0;
}
全部评论

相关推荐

今天周一休息,突发奇想写一篇阶段总结。如题,我已经去了一个和Java彻底毫无关联的行业。曾经我以为自己能在计算机行业发光发热,没想到刚入行一年多就当了逃兵。从最开始的热爱到现在一看到代码就厌恶,不知道自己经历了什么。所以我去干什么了?答案是:在成都当了租房销售。上班那会压力大了就念叨着去干租房中介,但是一直下不去这个决心,想着自己学了四年多的计算机知识,终究还是不甘心。终于在某一天准备八股文的时候,看着无数篇和工作内容关系不大的理论知识,那一刻下定决心,决定尝试一下销售行业,也算是给自己一个交代。后面阴差阳错的投了成都自如去当租房管家,没想到面试很顺利,在当天一百多个面试的人里面,我成为了为数不多通过的几个幸运儿之一。目前已经培训通过,正式入职,也开了单,也有压力但是每天过得很开心,真心喜欢那种和人交流的感觉,哪怕是最后没有选择找我租房。说这些也是想告诉那些大三,大四正在找Java实习而焦虑的同学:你们现在还年轻,选择很多,容错率也很高,可以尽情去尝试自己喜欢的行业和工作。不用因为某一次的面试没通过或者简历石沉大海而焦虑,更不用因为身边人都在挤编程的独木桥就强迫自己跟风。也算是自己的碎碎念吧,也希望自己能在新的领域取得一点小成就。也祝牛油工作顺利!
沉淀小子:干啥都不丢人啊,生存是必须要的,销售很考验一个人综合素质能力的,好的销售人脉和资源可不比写字楼的白领差啊
点赞 评论 收藏
分享
评论
2
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务