题解 | #小红送外卖#

小红送外卖

https://www.nowcoder.com/practice/2850d7c941f6494e82ba74bc899eb512

来去的路程是一样的,所以只需要跑一遍dijk就可以算出来,然后就可以计算答案了,注意是无向图就行。

#include <iostream>
#include <queue>
#include <map>
#include <set>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <iomanip>
#include <stack>
#include <numeric>
#include <ctime>
#include <string>
#include <bitset>
#include <unordered_map>
#include <unordered_set>

using namespace std;
using ll = long long;

const ll N = 5e5 + 5, mod = 1e9 + 7, inf = 2e18;
const double esp = 1e-8;

int n, m, q;

struct Node {
    ll v, w;
};

vector<Node>g[N];
bool vis[N];
ll dis[N];

struct node {
    ll id, vlu;
    bool operator<(const node& u)const {
        return u.vlu < vlu;
    }
};

void dijk(int s) {
    for (int i = 0; i <= n + 3; i++) {
        dis[i] = inf;
    }

    priority_queue<node>pq;

    pq.push({s, dis[s] = 0});

    while (!pq.empty()) {
        node tem = pq.top();
        pq.pop();

        if (vis[tem.id])continue;
        vis[tem.id] = true;

        for (auto [y, w] : g[tem.id]) {
            if (dis[y] > dis[tem.id] + w) {
                pq.push({y, dis[y] = dis[tem.id] + w});
            }
        }

    }
}

void solve() {
    cin >> n >> m >> q;

    while (m--) {
        ll u, v, w;
        cin >> u >> v >> w;
        g[u].push_back({v, w});
        g[v].push_back({u,w});
    }

    dijk(1);
    ll ans=0;
    while(q--){
        int x;
        cin>>x;
        ans+=2*dis[x];
    }
    cout<<ans<<'\n';
    // for(int i=1;i<=n;i++){
    //     cout<<dis[i]<<" \n"[i==n];
    // }
}

int main() {
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);

    int t = 1;
    //cin >> t;

    while (t--) {
        solve();
    }

    return 0;
}

全部评论

相关推荐

点赞 评论 收藏
分享
来,说点可能被同行“骂”的大实话。🙊当初接数字马力Offer时,朋友都说:“蚂蚁的“内包”公司?你想清楚啊!”但入职快一年后的今天,我反而对他有了不一样的看法!🔹&nbsp;是偏见?还是信息差!之前没入职之前外面都在说什么岗位低人一等这类。实际上:这种情况不可至否,不能保证每个团队都是其乐融融。但我在的部门以及我了解的周边同事都还是十分好相处的~和蚂蚁师兄师姐之间也经常开一些小玩笑。总之:身份是蚂蚁公司给的,地位是自己挣的(一个傲娇女孩的自述)。🔹&nbsp;待遇?玩的就是真实!试用期工资全额发!六点下班跑得快(早9晚6或者早10晚7,动态打卡),公积金顶格交。别听那些画饼的,到手的钱和下班的时间才是真的(都是牛马何必难为牛马)。🔹&nbsp;能不能学到技术?来了就“后悔”!我们拥有权限直通蚂蚁知识库,技术栈多到学不完。说“学不到东西”的人,来了可能后悔——后悔来晚了(哈哈哈哈,可以不学但是不能没有)!💥&nbsp;内推地址:https://app.mokahr.com/su/ueoyhg❗我的内推码:NTA6Nvs走我的内推,可以直达业务部门,面试流程更快速,进度可查!今天新放HC,之前挂过也能再战!秋招已经正式开始啦~机会就摆在这,敢不敢来试一试呢?(和我一样,做个勇敢的女孩)
下午吃泡馍:数字马力的薪资一般哇,5年经验的java/测试就给人一万出头,而且刚入职第三天就让人出差,而且是出半年
帮你内推|数字马力 校招
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

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