标签：CF545E Paths pre int Trees edges que push dis

[CF545E] Paths and Trees - 最短路

Description

给定一张带正权的无向图和一个源点，求边权和最小的最短路径树。

Solution

跑最短路的时候，转移时尽量让当前边的边权最小

记录一下前驱，最后连出来就是答案

#include <bits/stdc++.h>
using namespace std;

#define int long long
const int N = 1e6 + 5;

struct Edge
{
    int u, v, w, id;
};

struct Node
{
    int d;
    int p;
    bool operator<(const Node &rhs) const
    {
        return d > rhs.d;
    }
};

int n, m;
vector<int> g[N]; // 存的是边的 id
vector<Edge> edges;

int vis[N], dis[N], pre[N];

signed main()
{
    ios::sync_with_stdio(false);
    cin >> n >> m;
    for (int i = 1; i <= m; i++)
    {
        int u, v, w;
        cin >> u >> v >> w;
        g[u].push_back(edges.size());
        edges.push_back({u, v, w, i});
        g[v].push_back(edges.size());
        edges.push_back({v, u, w, i});
    }
    int s;
    cin >> s;
    priority_queue<Node> que;
    memset(dis, 0x3f, sizeof dis);
    dis[s] = 0;
    que.push({0, s});
    while (que.size())
    {
        auto [d, p] = que.top();
        que.pop();
        if (vis[p])
            continue;
        vis[p] = 1;
        for (auto i : g[p])
        {
            auto [_, q, w, id] = edges[i];
            if (dis[q] > dis[p] + w)
            {
                dis[q] = dis[p] + w;
                pre[q] = i;
                que.push({dis[q], q});
            }
            else if (dis[q] == dis[p] + w && w < edges[pre[q]].w)
            {
                pre[q] = i;
            }
        }
    }

    vector<int> ans;
    int ansval = 0;
    for (int i = 1; i <= n; i++)
    {
        if (i == s)
            continue;
        ans.push_back(edges[pre[i]].id);
        ansval += edges[pre[i]].w;
    }

    sort(ans.begin(), ans.end());

    cout << ansval << endl;
    for (auto i : ans)
        cout << i << " ";
}

标签：CF545E,Paths,pre,int,Trees,edges,que,push,dis
来源： https://www.cnblogs.com/mollnn/p/14452166.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。