Dijkstra堆优化算法（时间复杂度：mlogn）

Dijkstra——邻接表存储的堆优化算法

在明白原理后（上一篇讲过），发现朴素算法的时间复杂度太大，在找最短的距离需要枚举n个方案，但是在优先队列里面的小跟堆直接找堆顶就行。

优化思路：

用邻接表存图，然后用类似bfs宽搜的原理来更新各个点

代码：

#include <iostream>
#include <queue>
#include <vector>
#include <cstring>
using namespace std;
typedef pair<int,int> PII;		//为什么要用pair 因为排序的时候要按照距离排序
int n,m;
priority_queue<PII,vector<PII>,greater<PII>>heap;//小顶堆

const int N=200000;
int h[N],ne[N],idx,e[N];//邻接表存图，数组模拟链表
int w[N];
int dis[N];
bool st[N];
void add(int x,int y,int c)
{
    w[idx]=c;	
    e[idx]=y;	//e[]用来存下标
    ne[idx]=h[x];
    h[x]=idx++;   
}
int dijkstra()
{
    memset(dis,0x3f,sizeof(dis));	
    dis[1]=0;
    heap.push({0,1});		
    while(heap.size())
    {
        auto k=heap.top();
        heap.pop();
        int suffix=k.second,distance=k.first;	
        if(st[suffix])continue;			
            st[suffix]=true;
        for(int i=h[suffix];i!=-1;i=ne[i])	//走向当前可以连通的点
        {
            int j=e[i];
            if(dis[j]>distance+w[i])		
            {
                dis[j]=distance+w[i];
                heap.push({dis[j],j});
            }
        }
    }
    if(dis[n]==0x3f3f3f3f)return -1;
    return dis[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;
}