题目来源:http://acm.hdu.edu.cn/showproblem.php?pid=2544
最短路
Time Limit: 5000/1000 MS(Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 83889 Accepted Submission(s): 36292
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
Source
UESTC 6thProgramming Contest Online
-----------------------------------------------------
思路
之前那篇文章图论: Dijkstra算法是不是堆优化,那样做时间复杂度还是O(n^2), 并不是O(mlogn)
真正的堆优化的做法是维护一个优先队列作为d数组,每次更新d数组的时候不用删去旧的值,只需把新的值push进队列,根据队列的性质会自动把最小值放到最前面。
因此,vis数组(一个节点到起始点的路径是否确定)也是需要的。因为虽然每轮循环会把最小值pop出队列,但队列由于更新不删去旧的值,一个节点可能在队列中存在多份。
一个细节需要注意的是,循环开始前要vis[0]=1(0表示起始点),否则起始点又会进入队列。
-----------------------------------------------------
代码
// Dijkstra算法堆优化
#include<iostream>
#include<fstream>
#include<queue>
#include<cstring>
using namespace std;
const int NMAX = 105;
int mat[NMAX][NMAX] = {};
bool vis[NMAX] = {};
struct node {
int id, weight;
node(int i, int w)
{
id = i;
weight = w;
}
bool operator< (const node &b) const
{
return weight > b.weight;
}
};
int main()
{
int n,m,i,j,s1,s2,w,minv,minid;
while (cin >> n >> m)
{
if (n==0 && m==0)
{
break;
}
memset(mat, 0, sizeof(mat));
memset(vis, 0, sizeof(vis));
vis[0] = 1;
for (i=0; i<m; i++)
{
cin >> s1 >> s2 >> w;
mat[--s1][--s2] = w;
mat[s2][s1] = w;
}
priority_queue<node> q;
for (i=1; i<n; i++)
{
if (mat[0][i]!=0)
{
node nn(i,mat[0][i]);
q.push(nn);
}
}
for (i=0; i<n-1; i++)
{
node nn = q.top();
q.pop();
while (vis[nn.id])
{
nn = q.top();
q.pop();
}
minid = nn.id;
minv = nn.weight;
if (minid == n-1)
{
cout << minv << endl;
break;
}
vis[minid] = 1;
for (j=0; j<n; j++)
{
if (!vis[j] && mat[minid][j]!=0)
{
nn.id = j;
nn.weight = minv + mat[minid][j];
q.push(nn);
}
}
}
}
return 0;
}