今日总结 – 【图论-最短路】Dijkstua堆优化

在找源点最近的点是选择用堆,每次取出堆顶,减少每次查找最近点的时间复杂度,用内嵌for循环时间复杂度时O(n^2),而如果用堆的话,时间复杂度就是O(nlogn)

下面就把代码放出来,存图用的是链式前向星,如果对链式前向星不太熟悉,可以去看一看我之前的帖子

下面是代码:

/********************************************************************************
* @author: Wu Zonglin
* @email: [email protected]
* @date: 2022/2/13 22:39
* @version: 1.0
* @description:
********************************************************************************/
#include <bits/stdc++.h>
#include <vector>
#define INF 2147483647
using namespace std;
typedef struct edge{
    long long v,w,next;
}E;
vector<E> e(50005,{0,0,-1});

vector<int> head(10005,-1);

typedef pair<int,int> pii;//存的是源点到某一点的距离
priority_queue<pii,vector<pii>,greater<pii>>min_heap;//建立优先队列 小根堆
void add_edge(int u,int v,int w){
    e.push_back({v, w, head[u]});
    head[u]=e.size()-1;
}
long long dj[10005];
bool book[10005];
int main(){
    int n,m,si;
    cin>>n>>m>>si;
    for (int i = 0; i < m; ++i) {
        int u,v,w;
        cin>>u>>v>>w;
        add_edge(u,v,w);
    }
    for (int i = 0; i < 10001; ++i) {
        dj[i]=INF;
    }
    min_heap.push(make_pair(0,si));
    dj[si]=0;
//    book[si]= true;
//    for (int i = head[si]; i+1; i=e[i].next) {
//        dj[e[i].v]=e[i].w;
//    }
    while(!min_heap.empty()) {
        int u=min_heap.top().second;
        min_heap.pop();
        book[u]=true;
        for (int j = head[u]; j+1 ; j=e[j].next) {
            int v=e[j].v;
            if(!book[v]&&dj[v]>=dj[u]+e[j].w){
                dj[v]=dj[u]+e[j].w;
                min_heap.push(make_pair(dj[v],v));
            }
        }
    }
    for (int i = 1; i <= n; ++i) {
        cout<<dj[i]<<" ";
    }
}

正文完