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;
}