dijkstra算法的堆优化版,用了一个pair对来存距离和链表中的序号,然后按照模板写下来就ok了。
就存距离的时候要注意下标要减1来存,因为存图的时候idx已经++了。
#include <bits/stdc++.h>
using namespace std;
const int N = 10000000;
typedef pair<int, int> PII; //分别存距离和序号
int n,m,s;
int h[N], w[N], e[N], ne[N], idx ;
int dist[N];
bool st[N];
void add(int a, int b)
{
e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
return;
} //邻接表标准存储写法
void dijkstra()
{
memset(dist,0x3f, sizeof dist); //初始化到无穷大
dist[s] = 0; //起点距离为0
priority_queue<PII, vector<PII>, greater<PII>> heap;
heap.push({0, s});
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] > distance + w[i])
{
dist[j] = distance + w[i];
heap.push({dist[j], j});
}
}
}
return;
}
int main()
{
memset(h, -1, sizeof h);
cin>>n>>m>>s;
for(int i=1;i<=m;i++)
{
int a,b,c;;
cin>>a>>b>>c;
add(a,b);
w[idx-1]=c; //这时候idx已经加1了,所以存距离需要减1
}
dijkstra();
for(int i=1;i<=n;i++)
{
cout<<dist[i];
if(i!=n)cout<<' ';
}
return 0;
}