堆优化Dijkstra算法求最短路 并输出最短的路径

如输入:
5 7
1 2 5
3 1 4
2 4 8
3 2 3
5 2 9
3 4 7
4 5 6
输出:
14
1 2 5

#include<bits/stdc++.h>

using  namespace std;

const int N = 1010, M = 2e4+10;
typedef pair<int, int> pii;
#define x first
#define y second
#define inf 0x3f3f3f3f
int dist[N], h[N], e[M], ne[M], w[M], idx;
bool st[N];
int n, p;
int Prev[N];

void add(int a, int b, int c)
{
        e[idx] = b, ne[idx] = h[a], w[idx] = c, h[a] = idx++;
}

int dijkstra(int sta)
{
        memset(dist, 0x3f, sizeof dist);
        memset(st, false, sizeof st);
        dist[sta] = 0;
        priority_queue<pii, vector<pii>, greater<pii>> pq;
        pq.push({dist[sta], sta});
        while(pq.size())
        {
                pii tmp = pq.top();
                pq.pop();
                int ver = tmp.y, distance = tmp.x;
                if(st[ver]) continue;
                st[ver] = true;
                for(int i=h[ver]; ~i; i=ne[i]){
                        int j = e[i];
                        if(dist[j]>distance + w[i]){
                                Prev[j] = ver;
                                dist[j] = distance + w[i];
                                pq.push({dist[j], j});
                        }
                }
        }
        if(dist[1]==inf) return inf;
        return dist[1];
}

int main()
{
        cin>>n>>p;
        memset(h, -1, sizeof h);
        for(int i=0;i<p;++i)
        {
                int a, b, w;
                cin>>a>>b>>w;
                add(a, b, w), add(b, a, w);
        }

        int t = dijkstra(n);
        if(t==inf) {
                cout<<-1<<endl;
                return 0;
        }else{
                cout<<t<<endl;
                int tmp1, tmp2;
                for(int i=0; !(tmp2 == n); ++i)
                {
                        if(i==0) tmp1 = 1, cout<<1<<' ';
                        else
                        {
                                tmp2 = Prev[tmp1];
                                cout<<tmp2<<' ';
                                tmp1 = tmp2;
                        }
                }
        }


        return 0;
}


  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值