hdu(2544)——最短路(邻接表+bellman-ford使用队列优化)

#include<cstdio>
#include<cstring>
#include<map>
#include<set>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
#include<iostream>
#include<time.h>
using namespace std;
typedef __int64 ll;
typedef unsigned __int64 ULL;
#define pi acos(-1.0)
#define Ex exp(1.0)
#define inf 1000000000
#define maxn 10010
int vis[maxn],dis[maxn],first[maxn],next1[maxn*2];
int e;
queue<int> que;
struct node{
int u,v,w;
node() {}
node(int u,int v,int w):u(u),v(v),w(w){}
}p[maxn*2];
void init(){
memset(p,0,sizeof(p));
memset(first,-1,sizeof(first));
memset(next1,-1,sizeof(next1));
}
void addnote(int u,int v,int w){
p[e]=node(u,v,w);
next1[e]=first[u]; first[u]=e++;
}
int main(){
int n,m;
while(~scanf("%d%d",&n,&m)){
if(n==0&&m==0) break;
init();
e=1;
for(int i=0;i<m;i++){
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
addnote(u,v,w);
addnote(v,u,w);
}
while(!que.empty()) que.pop();
for(int i=0;i<110;i++) dis[i]=inf;
dis[1]=0;
memset(vis,0,sizeof(vis));
que.push(1); vis[1]=1;
while(!que.empty()){
int idx=que.front();
int k=first[idx];
vis[idx]=0;     //一开始这里没有注意到
//因为是无向图，所以要取消标记，有可能后面更新的，从而也能使前面的点得以更新
//所以每次都要进行取消标记的操作！！
while(k!=-1){
if(dis[p[k].v]>dis[p[k].u]+p[k].w){
dis[p[k].v]=dis[p[k].u]+p[k].w;
if(!vis[p[k].v]){
vis[p[k].v]=1;
que.push(p[k].v);
}
}
k=next1[k];
}
que.pop();
}
printf("%d\n",dis
);
}
return 0;
}
/*
5 5
1 2 10
1 3 4
2 4 2
3 4 3
2 5 5
*/