【最短路】迪杰斯特拉算法（朴素法以及堆+邻接表优化版）

题目

 活动 - AcWing

解释

AcWing 849. Dijkstra求最短路 I - AcWing

• g[i][j]代表i到j的距离
• 用上界0x3f3f3f3f(在memset里可以用0x3f来替代）初始化没有边的两个结点
• 执行n-1次循环，每次确定一个点到1号结点的最短距离
• 每次循环中查找没有确定最短距离的结点中离1号结点距离最近的那个结点dist[t]
• 然后比较这些没有确定最短距离的结点的当前dist和dist[t]+g[t][i]的大小
• 为什么可以这么做需要查找具体的迪杰斯特拉算法数学证明
• 最外层for循环执行n-1次是因为，最后一个结点可能要经过n-1个结点才是最短距离
• 也可以理解成每执行一次确定一个结点的最短距离，这样至少确定完n-1个结点的最短距离
• 此时第n个结点的最短距离也就出来了

代码段

#include<bits/stdc++.h>
using namespace std;
int n,m;
const int N=510;
int g[N][N];
int dist[N];
bool st[N];
int dijkstra()
{
    memset(dist,0x3f,sizeof dist);
    dist[1]=0;
    for(int i=1;i<=n-1;i++)//执行n-1次,这里只要大于等于n-1次都可以
    {
        int t=-1;
        for(int j=1;j<=n;j++)
            if(!st[j]&&(t==-1||dist[t]>dist[j]))
            t=j;
            for(int j=1;j<=n;j++)
           if(!st[j])
           dist[j]=min(dist[j],dist[t]+g[t][j]);
        
        st[t]=true;
    }
    if(dist[n]==0x3f3f3f3f)return -1;
    else
    return dist[n];
}
int main()
{
    cin>>n>>m;//n个点m个边
    memset(g,0x3f,sizeof g);
    while(m--)
    {
        int a,b,c;
        cin>>a>>b>>c;
        g[a][b]=min(g[a][b],c);
    }
    cout<<dijkstra()<<endl;
}

堆+邻接表优化

解释

AcWing 850. 朴素Dijkstra与堆优化Dijkstra总结 - AcWing

• 遗留问题在对于邻接表的实现

代码段

#include<bits/stdc++.h>
using namespace std;
#define pii pair<int,int>
const int N=1e6+10;
int dist[N];
bool st[N];
int h[N],e[N],ne[N],w[N],idx;
int n,m;
void add(int x,int y,int c)
{
    w[idx]=c;
    e[idx]=y;
    ne[idx]=h[x];
    h[x]=idx++;
}//idx应该看成是一个边,h[x]数组代表，以x为开头的边是几号（idx）边
//ne[idx]代表,以y结尾的idx号边的下一个边是多少，如果没有边了那么就是-1
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())
    {
        pii k=heap.top();
        heap.pop();
        int ver=k.second,distance =k.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]>distance+w[i])
            {
                dist[j]=distance+w[i];//这里的w[i]即e[i]到ver的距离（idx边）
                heap.push({dist[j],j});
            }
        }
    }
    if(dist[n]==0x3f3f3f3f)return -1;
    else return dist[n];
}
int main()
{
    memset(h,-1,sizeof h);
    cin>>n>>m;
    while(m--)
    {
        int x,y,c;
        cin>>x>>y>>c;
        add(x,y,c);
    }
    cout<<dijkstra()<<endl;
    return 0;
}