Dijkstra求最短路径

849. Dijkstra求最短路 I

原题链接

代码

朴素版

#include<iostream>
#include<cstring>
using namespace std;
const int N=510;
int dist[N],g[N][N],n,m;
bool st[N];
int dijkstra(){
    memset(dist,0x3f,sizeof dist);
    dist[1]=0;
    for(int i=0;i<n-1;i++){//找n-1次，因为第一个点到自己的距离是零 只用对剩下的n-1个点处理就好了
        int t=-1;
        for(int j=1;j<=n;j++){//循环n次，找到距离最短的点
            if(!st[j]&&(t==-1||dist[t]>dist[j]))
                t=j;
        }
        for(int j=1;j<=n;j++)
            dist[j]=min(dist[j],dist[t]+g[t][j]);//循环n次，更新每个点的最短距离
        st[t]=true;
    }
    if(dist[n]==0x3f3f3f3f)return -1;
    return dist[n];
    
}

int main(){
    memset(g,0x3f,sizeof g);
    cin>>n>>m;
    while(m--){
        int a,b,w;
        cin>>a>>b>>w;
        g[a][b]=min(w,g[a][b]);
    }
    cout<<dijkstra()<<endl;
    return 0;
}

850. Dijkstra求最短路 II

堆优化版

#include<iostream>
#include<queue>
#include<cstring>
using namespace std;
typedef pair<int,int> PII;
const int N=1e6+10;
int e[N],ne[N],h[N],w[N],idx;
int dist[N],n,m;
bool st[N];
void add(int a,int b,int c){
    e[idx]=b;
    w[idx]=c;
    ne[idx]=h[a];
    h[a]=idx++;
}
int dijkstra(){
    memset(dist,0x3f,sizeof dist);
    dist[1]=0;
    priority_queue<PII, vector<PII>, greater<PII>> heap;
    heap.push({0,1});
    while(heap.size()){
        auto t=heap.top();
        heap.pop();
        int ver=t.second,distance=t.first;
        if(st[ver])continue;
        st[ver]=true;
        for(int i=h[ver];i!=-1;i=ne[i]){
            int j=e[i];
            if(dist[j]>dist[ver]+w[i]){
                dist[j]=dist[ver]+w[i];
                heap.push({dist[j],j});
            }
                
        }
    }
    if(dist[n]==0x3f3f3f3f)return -1;
    return dist[n];
    
}

int main(){
    memset(h,-1,sizeof h);
    cin>>n>>m;
    while(m--){
        int a,b,c;
        cin>>a>>b>>c;
        add(a,b,c);
    }
    cout<<dijkstra()<<endl;
    return 0;
}