POJ - 2387 dijkstra算法 链式前向星+堆优化

题目链接

求顶点1到n的最短路     

dijkstra 链式前向星+堆优化的板子

#include<algorithm>
#include <iostream>
#include<cstring>
#include<vector>
#include<queue>
#include <cstdio>
using namespace std;
typedef long long ll;
const int INF=0x3f3f3f3f;
const int maxn=1005;
struct node{
    int to,w,nexts;
}edge[4*maxn];
int d[maxn];
typedef pair<int,int> p;
int head[maxn];
int n,m;
int cnt=0;
void add(int i,int j,int w){
    edge[++cnt].to=j;
    edge[cnt].w=w;
    edge[cnt].nexts=head[i];
    head[i]=cnt;
}
void dijkstra(int s){
    priority_queue<p,vector<p>,greater<p> > q;
    fill(d,d+n+1,INF);
    d[s]=0;
    q.push(p(0,s));
    while(!q.empty()){
        p pp=q.top();q.pop();
        int v=pp.second;
        if(d[v]<pp.first) continue;
        for(int i=head[v];i;i=edge[i].nexts){
            node e=edge[i];
            if(d[e.to]>d[v]+e.w){
                d[e.to]=d[v]+e.w;
                q.push(p(d[e.to],e.to));
            }
        }
    }
}
int main(){
    scanf("%d%d",&m,&n);
    memset(head,0,sizeof(head));
    for(int i=0,a,b,w;i<m;i++){
        scanf("%d%d%d",&a,&b,&w);
        add(a,b,w);
        add(b,a,w);
    }
    dijkstra(1);
    printf("%d\n",d[n]);
    return 0;
}