模板题用来练手。
现在来说一般的图论题目都很难用普通dijkstra过掉,而SPFA又很不稳定,还是学了一下国际公认的堆优化dijkstra。
简单来说,堆优化dij就是把for循环找最小的d[i]那维用堆来做,将O(n)降成了O(logn)。
关于一个小问题见程序注释
【据说priority_queue常数巨大 ,不管了】
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <queue>
#include <vector>
using namespace std;
const int N=10005;
int node[N],nxt[N],head[N],data[N];
int n,m,d[N],tot;
int i,j;
void add(int x,int y,int z){
node[++tot]=y;nxt[tot]=head[x];head[x]=tot;data[tot]=z;
}
void init(){
tot=0;
memset(head,0,sizeof head);
memset(nxt,0,sizeof nxt);
memset(node,0,sizeof node);
for (i=1;i<=m;i++){
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
add(x,y,z);add(y,x,z);
}
}
void Dij(){
memset(d,10,sizeof d);d[1]=0;
priority_queue<pair<int,int> > heap;
heap.push(make_pair(-d[1],1));
while (1){
for (;!heap.empty() && -d[heap.top().second]!=heap.top().first;heap.pop());
//对于每个d[i],只会被更新的越来越小,所以如果当前的堆顶的原本的d[]不等于现在的d[],说明肯定已经被更新过,不是最优的
if (heap.empty()) break; //一开始没加这句话 错的非常惨。。。。 调了一整天。。。。
int now=heap.top().second;
heap.pop();
for (i=head[now];i;i=nxt[i]){
j=node[i];
if (d[j]>d[now]+data[i]){
d[j]=d[now]+data[i];
heap.push(make_pair(-d[j],j));
}
}
}
}
int main(){
for (scanf("%d%d",&n,&m);n && m;scanf("%d%d",&n,&m)){
init();
Dij();
printf("%d\n",d[n]);
}
return 0;
}