有权图的单源最短路算法 优先级队列优化dij

17 篇文章 0 订阅
2 篇文章 0 订阅

思想是:将未被收录的点放入一个优先级队列当中,每次队首元素就是这次要收录的点

测试样例


#include <iostream>
#include <queue>
#include <stdio.h>
#include <vector>
using namespace std;
struct node
{
    int num,dis;
    node(int nn,int dd):num(nn),dis(dd) {}
    friend bool operator <(const node &a,const node &b)
    {
        if(a.dis==b.dis)
            return a.num<b.num;
        else
            return a.dis>b.dis;
    }
};
int N,R;
int dist[105];
int path[105];
const int IN = 1<<30;
vector<node> G[105];
priority_queue<node> q;
int MinDis;
int dij()
{
    MinDis = IN;
    for(int i=1; i<=N; i++)
        dist[i] = IN;
    dist[1] = 0;
    q.push(node(1,0));
    while(!q.empty())
    {
        node now = q.top();
        q.pop();
        for(int i=0; i<G[now.num].size(); i++)
        {
            node p = G[now.num][i];
            if(now.dis + p.dis <dist[p.num])
                {
                    path[p.num] = now.num;
                    dist[p.num] = now.dis + p.dis;
                    q.push(node(p.num,dist[p.num]));
                }
        }
    }
}
int main()
{
    cin >> N >> R;
    int source,desin,dis;
    while(R--)
    {
        cin >> source >> desin >> dis;
        G[source].push_back(node(desin,dis));
        G[desin].push_back(node(source,dis));
    }
    dij();
    for(int i=1;i<=N;i++)
        cout<<dist[i]<<endl;
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值