以poj2387为例
dijstra的思想是每次在dis数组里面找出最小的那一条边,然后用这一条边去松弛其他的边,在这里我们把dis数组保存在一个优先队列里面,同时把图存在邻接表里面,每次通过优先队列得出dis数组里面最小的那条边,然后通过邻接表去松弛与这条边相连的其他的边,然后把更新的dis数组存入优先对列,直到队列为空时结束。
ac代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<string>
#include<algorithm>
#include<vector>
#include<set>
#include<cstdlib>
#include<map>
#include<deque>
#include<queue>
#include<list>
const int inf=0x3f3f3f3f;
const int MOD=1e9+7;
const int inff=999999;
const int N=1005;
const int M=2005;
#define LL long long
#define ME0(x) memset(x,0,sizeof(x))
#define MEF(x) memset(x,-1,sizeof(x))
#define MEI(x) memset(x,inf,sizeof(x))
using namespace std;
int n,m,cnt;
struct LX
{
int ed,di,next;
}lx[M*2];
int first[N];
int dis[N],vis[N];
void add(int s,int e,int d)
{
cnt++;
lx[cnt].ed=e;
lx[cnt].di=d;
lx[cnt].next=first[s];
first[s]=cnt;
}
struct cmp
{
bool operator ()(const int a,const int b)
{
return dis[a]>dis[b];
}
};
int dijstra(int s)
{
priority_queue<int,vector<int>,cmp> pq;
for(int n1=1;n1<=n;n1++)
{
dis[n1]=inff;
}
dis[s]=0;
ME0(vis);
pq.push(s);
while(!pq.empty())
{
int xe=pq.top();
pq.pop();
if(vis[xe])
{
continue;
}
vis[xe]=1;
for(int i=first[xe];i;i=lx[i].next)
{
int ie=lx[i].ed;
int di=lx[i].di;
if(!vis[ie]&&dis[ie]>dis[xe]+di)
{
dis[ie]=dis[xe]+di;
pq.push(ie);
}
}
}
return dis[n];
}
int main()
{
int s,e,d;
scanf("%d%d",&m,&n);
cnt=0;
ME0(first);
for(int m1=1;m1<=m;m1++)
{
scanf("%d%d%d",&s,&e,&d);
add(s,e,d);
add(e,s,d);
}
printf("%d\n",dijstra(1));
}