Dijkstra(邻接表实现)

 

#include<iostream>
#include<string.h>
#include<vector>
#include<queue>
using namespace std;
#define INF 0x3f3f3f3f
#define Max_N 1000
struct edge{int to,cost;};
typedef pair<int,int> P;  //first为到S最短距离,second为顶点编号

int V,E;
vector<edge> G[Max_N];  //每个G[v]里面包含一个边的edge类型vector,记录与顶点v相连的点和他们组成边的权值
int d[Max_N];

void dijkstra(int s)
{
    //创造优先队列que
    priority_queue<P,vector<P>,greater<P> > que;//最后两个>间有同一个空格,对于pair的P,默认first为排序依据
    fill(d,d+V,INF);
    d[s]=0;
    que.push(P(0,s));

    while(!que.empty())
    {
        P p=que.top();
        que.pop();
        int v=p.second;
        if(d[v]<p.first) continue;  //原来的值更小,不更新,跳过
        for(int i=0;i<G[v].size();i++)   //遍历与顶点v相连的边
        {
            edge e=G[v][i];
            if(d[e.to]>d[v]+e.cost)
            {
                d[e.to]=d[v]+e.cost;
                que.push(P(d[e.to],e.to));
            }
        }
    }
}
int main()
{
    //freopen("dijkstra.txt","r",stdin);
    //录入有向权图,起点S默认为第一个录入顶点
    cin>>V>>E;
    for(int i=0;i<E;i++)
    {
        int s,t,co;  //s为起点,t为终点,co为权
        cin>>s>>t>>co;
        G[s].push_back({t,co});
    }
    //
    dijkstra(0);
    for(int i=1;i<V;i++)
        cout<<"S到序号"<<i<<"顶点的最短距离:"<<d[i]<<endl;

}

 

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