HDU 2544：最短路

最短路

Time Limit: 5000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 81379    Accepted Submission(s): 35232

Problem Description

在每年的校赛里，所有进入决赛的同学都会获得一件很漂亮的t-shirt。但是每当我们的工作人员把上百件的衣服从商店运回到赛场的时候，却是非常累的！所以现在他们想要寻找最短的从商店到赛场的路线，你可以帮助他们吗？

Input

输入包括多组数据。每组数据第一行是两个整数N、M（N<=100，M<=10000），N表示成都的大街上有几个路口，标号为1的路口是商店所在地，标号为N的路口是赛场所在地，M则表示在成都有几条路。N=M=0表示输入结束。接下来M行，每行包括3个整数A，B，C（1<=A,B<=N,1<=C<=1000）,表示在路口A与路口B之间有一条路，我们的工作人员需要C分钟的时间走过这条路。
输入保证至少存在1条商店到赛场的路线。

Output

对于每组输入，输出一行，表示工作人员从商店走到赛场的最短时间

Sample Input

2 1
1 2 3
3 3
1 2 5
2 3 5
3 1 2
0 0

Sample Output

3
2

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>
#define INF 0x3f3f3f3f
const int maxn=1e3+10;
int way[maxn][maxn];
int flag[maxn],dis[maxn];
int n,m;
int a,b,c;
void Dijkstra()
{
	memset(flag,0,sizeof(flag));
	int ans,x;
	for(int i=1;i<=n;i++) dis[i]=way[1][i];
	dis[1]=0;
	flag[1]=1;
	for(int i=1;i<=n;i++)
	{
		ans=INF;
		for(int j=1;j<=n;j++)
		{
			if(ans>dis[j]&&!flag[j])
			{
				ans=dis[j];
				x=j;
			}
		}
		flag[x]=1;
		for(int j=1;j<=n;j++)
		if(!flag[j])
			dis[j]=std::min(dis[j],dis[x]+way[x][j]);
	}
}
int main()
{
	while(~scanf("%d%d",&n,&m))
	{
		if(n==0&&m==0) break;
		for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
		way[i][j]=INF;
		while(m--)
		{
			scanf("%d%d%d",&a,&b,&c);
			way[a][b]=way[b][a]=c;
		}
		Dijkstra();
		printf("%d\n",dis[n]);
	}
	return 0;
} 

链式前向星+堆优化写法：

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <limits.h>
#include <map>
#include <stack>
#include <queue>
#include <vector>
#include <set>
#include <string>
#define ll long long
#define ull unsigned long long
#define ms(a) memset(a,0,sizeof(a))
#define msy(a) memset(a,-1,sizeof(a))
#define msf(a) memset(a,INF,sizeof(a))
#define pi acos(-1.0)
#define INF 0x7f7f7f7f
#define lson o<<1
#define rson o<<1|1
const double E=exp(1);
const int maxn=1e6+10;
const int mod=1e9+7;
using namespace std;
struct Edge
{
	int Next;	//下一条变的储存下标
	int to;		//这条边的终点
	int w;		//权值
}edge[maxn];
int cnt;
int n,m;
// head数组初始化为-1
int head[maxn];	//head[i]表示以i为起点的第一条边
int dis[maxn];
void add(int u,int v,int w)	//起点u，终点v，权值w
{
	edge[cnt].Next=head[u];
	edge[cnt].w=w;
	edge[cnt].to=v;
	head[u]=cnt++;	//第一条边为当前边
}
// 遍历以u为起始位置的所有边(逆序遍历)
// for(int i=head[u];~i;i=edge[i].Next)
struct Rule
{
	bool operator()(int &a,int &b) const{return dis[a]>dis[b];}
};
void dijkstra(int a)
{
	msf(dis);
	priority_queue<int,vector<int>,Rule>q;
	dis[a]=0;
	q.push(a);
	while(!q.empty())
	{
		int u=q.top();q.pop();
		for(int k=head[u];~k;k=edge[k].Next)
		{
			int &v=edge[k].to;
			if(dis[v]>dis[u]+edge[k].w)
			{
				dis[v]=dis[u]+edge[k].w;
				q.push(v);
			}
		}
	}
}
int main(int argc, char const *argv[])
{
	ios::sync_with_stdio(false);
	while(cin>>n>>m&&n&&m)
	{
		cnt=0;
		int x,y,z;
		msy(head);
		// cin>>n>>m;
		for(int i=1;i<=m;i++)
		{
			cin>>x>>y>>z;
			// 无向图add两次，有向图一次
			add(x,y,z);
			add(y,x,z);
		}
		dijkstra(1);
		if(dis[n]==INF)
			cout<<-1<<endl;
		else
			cout<<dis[n]<<endl;	
	}
	return 0;
}