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

#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]);
    }
}