HDU 2544 - 最短路 - [堆优化dijkstra][最短路模板题]

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2544

Time Limit: 5000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)

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

AC代码:

#include<bits/stdc++.h>
using namespace std;
const int maxn=110;
const int INF=0x3f3f3f3f;

int n,m;

struct Edge
{
    int u,v,w;
    Edge(int u,int v,int w){this->u=u,this->v=v,this->w=w;}
};
vector<Edge> E;
vector<int> G[maxn];
void init(int l,int r)
{
    E.clear();
    for(int i=l;i<=r;i++) G[i].clear();
}
void addedge(int u,int v,int w)
{
    E.push_back(Edge(u,v,w));
    G[u].push_back(E.size()-1);
}

bool vis[maxn];
int d[maxn];
void dijkstra(int st)
{
    for(int i=1;i<=n;i++) d[i]=(i==st)?0:INF;
    memset(vis,0,sizeof(vis));

    priority_queue< pair<int,int> > Q;
    Q.push(make_pair(0,st));
    while(!Q.empty())
    {
        int now=Q.top().second; Q.pop();
        if(vis[now]) continue;
        vis[now]=1;
        for(int i=0;i<G[now].size();i++)
        {
            Edge &e=E[G[now][i]]; int nxt=e.v;
            if(vis[nxt]) continue;
            if(d[nxt]>d[now]+e.w)
            {
                d[nxt]=d[now]+e.w;
                Q.push(make_pair(-d[nxt],nxt));
            }
        }
    }
}

int main()
{
    while(scanf("%d%d",&n,&m) && n+m>0)
    {
        init(1,n);
        for(int i=1,u,v,w;i<=m;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            addedge(u,v,w);
            addedge(v,u,w);
        }

        dijkstra(1);
        printf("%d\n",d[n]);
    }
}

猜你喜欢

转载自www.cnblogs.com/dilthey/p/9011700.html