STL 优先队列实现的DIJSKTRA算法

时间:2022-04-24 11:00:03

   P转C几个月了,还在坚持用C写P的代码。。。是时候学习一下C++的STL库了。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>
#include <queue>
#include <utility>
using namespace std;
const int MAX=100001,INF = 1000000000;
typedef pair<int,int> pii;
vector<pii> a[MAX];
int n,d[MAX];
bool done[MAX];
priority_queue< pii,vector<pii>,greater<pii> > q;
int main()
{
for(int i=1;i <= n;i++) d[i]=(i == 1 ? 0 : INF);
memset(done,0,sizeof(done));
q.push(make_pair(d[1],1));
while(!q.empty())
{
int u=q.top().second;
q.pop();
if(done[u]) continue;
done[u]=true;
for(pii v:a[u])
{
if(d[v.first] > d[u]+v.second)
{
d[v.first]=d[u]+v.second;
q.push(make_pair(d[v.first],v.first));
}
}
}
}