loj 1099(最短路）

题目链接：http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=25956

思路：dist[v][0]代表走到点v的最短路，dist[v][1]代表走到点v的次短路（dist[v][0]!=dist[v][1]),然后Dijkstra更新就可以了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
#define MAXN 5555
#define MAXM 555555
#define inf 1<<30
#define FILL(a,b) memset(a,b,sizeof(a))

struct Edge{
    int v,w,next;
}edge[MAXM];

int n,m,NE;
int head[MAXN];

void Insert(int u,int v,int w)
{
    edge[NE].v=v;
    edge[NE].w=w;
    edge[NE].next=head[u];
    head[u]=NE++;
}

int dist[MAXN][2];
int Dijkstra()
{
    for(int i=1;i<=n;i++)
        for(int j=0;j<=1;j++)dist[i][j]=inf;
    priority_queue<pair<int,int>,vector<pair<int,int> >,greater<pair<int,int> > >que;
    que.push(make_pair(0,1));
    dist[1][0]=0;
    while(!que.empty()){
        pair<int,int>p=que.top();
        que.pop();
        for(int i=head[p.second];i!=-1;i=edge[i].next){
            int v=edge[i].v,w=edge[i].w;
            if(p.first+w<dist[v][0]){
                dist[v][1]=dist[v][0];
                dist[v][0]=p.first+w;
                que.push(make_pair(dist[v][0],v));
            }else if(p.first+w!=dist[v][0]&&p.first+w<dist[v][1]){
                dist[v][1]=p.first+w;
                que.push(make_pair(dist[v][1],v));
            }
        }
    }
    return dist[n][1];
}
int main()
{
    int _case,t=1;
    scanf("%d",&_case);
    while(_case--){
        scanf("%d%d",&n,&m);
        NE=0;
        FILL(head,-1);
        while(m--){
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            Insert(u,v,w);
            Insert(v,u,w);
        }
        printf("Case %d: %d
",t++,Dijkstra());
    }
    return 0;
}