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

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

测试样例


#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;
}

阅读更多
个人分类: dij 最短路
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

不良信息举报

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

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭