Light oj 1099 - Not the Best 次短路

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#define MAXSIZE 500005
#define INF 0x3f3f3f3f
#define LL long long

using namespace std;

int ans,n,k,a[MAXSIZE],dist1[MAXSIZE],dist2[MAXSIZE],vis[MAXSIZE];
typedef pair<int, int>p;//p.second储存节点：u，p.frist储存s->u的当前最短距离

struct node
{
    int u;
    int v;
    int w;
    int next;
}G[MAXSIZE];

void Add(int u,int v,int w)
{
    G[k].u=u;
    G[k].v=v;
    G[k].w=w;
    G[k].next=a[u];
    a[u]=k++;
}

void dfs()
{
    priority_queue<p,vector<p>,greater<p> > Q;
    dist1[1]=0;
    Q.push(p(0,1));
    while(!Q.empty())
    {
        p k=Q.top();
        Q.pop();
        int u=k.second;
        int d=k.first;
        if(dist2[u] < d) //小优化：如果取出的不是最短距离就不再向下进行
           continue;
        for(int i=a[u];i!=-1;i=G[i].next)
        {
            int v=G[i].v;
            int d2=d+G[i].w;
            if(dist1[v] > d2) //更新最短路
            {
                swap(dist1[v],d2);
                Q.push(p(dist1[v],v));
            }
            if(dist2[v] > d2 && dist1[v]<d2)//更新次短路
            {
                dist2[v]=d2;
                Q.push(p(dist2[v],v));
            }
        }
    }
    ans=dist2[n];
}

void Init()
{
    for(int i=0;i<MAXSIZE;i++)
    {
        dist1[i]=INF;
        dist2[i]=INF;
        vis[i]=0;
        a[i]=-1;
    }
    k=1;
}

int main()
{
    int T,cns=1,u,v,w,m;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        Init();
        while(m--)
        {
            scanf("%d%d%d",&u,&v,&w);
            Add(u,v,w);
            Add(v,u,w);
        }
        dfs();
        printf("Case %d: %d\n",cns++,ans);
    }
    return 0;
}