HDU 3986 Harry Potter and the Final Battle

跟HDU-1595完全一样，稍改一下输出就好了……

#include <algorithm>
#include <iostream>
#include <fstream>
#include <sstream>
#include <iomanip>

#include <map>
#include <set>
#include <list>
#include <stack>
#include <queue>
#include <deque>
#include <vector>
#include <string>
#include <bitset>
#include <memory>
#include <complex>
#include <numeric>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <time.h>
#include <ctype.h>
#include <locale.h>

using namespace std;

#pragma pack(4)

const double  eps = 1e-8;
const double   pi = acos(-1.0);
const int     inf = 0x7f7f7f7f;
const __int64 INF = 0x7f7f7f7f7f7f7f7fll;

#define   at(a,i) ((a)&(1<<(i)))
#define   nt(a,i) ((a)^(1<<(i)))
#define set1(a,i) ((a)|(1<<(i)))
#define set0(a,i) ((a)&(~(1<<(i))))

#define gret(a,b) (((a)-(b))>eps)
#define less(a,b) (((b)-(a))>eps)
#define greq(a,b) (((a)-(b))>-eps)
#define leeq(a,b) (((b)-(a))>-eps)
#define equl(a,b) (fabs((a)-(b))<eps)

#define lmax(a,b) ((a)>(b)?(a):(b))
#define lmin(a,b) ((a)<(b)?(a):(b))
#define fmax(a,b) (gret(a,b)?(a):(b))
#define fmin(a,b) (less(a,b)?(a):(b))

const int MAXV = 1200;
const int MAXE = 120000;

struct node
{
    int v,w;
}G[MAXE];
int _index,pre[MAXV],next[MAXE];

void clear(void)
{
    _index=0;
    memset(pre,-1,sizeof(pre));
}
void add(int u,int v,int w)
{
    G[_index].v=v;
    G[_index].w=w;
    next[_index]=pre[u];
    pre[u]=_index++;
}

bool inQ[MAXV];
int Q[MAXV],dis[MAXV],prev[MAXV],curr[MAXV];

int SPFA(int src,int des)
{
    int u,v,w;
    memset(dis,inf,sizeof(dis));
    memset(inQ,false,sizeof(inQ)); dis[Q[0]=src]=0;

    for(int f=0,r=1;f!=r;f=(f+1)%MAXV)
    {
        inQ[u=Q[f]]=false;
        for(int i=pre[u];i!=-1;i=next[i])
        {
            v=G[i].v;
            w=G[i].w;
            if(dis[v]>dis[u]+w)
            {
                dis[v]=dis[u]+w;
                prev[v]=u;
                curr[v]=i;

                if(!inQ[v])
                {
                    inQ[Q[r]=v]=true;
                    r=(r+1)%MAXV;
                }
            }
        }
    }

    return dis[des];
}
int findAns(int src,int des,int k_th)
{
    int u,v,w;
    memset(dis,inf,sizeof(dis));
    memset(inQ,false,sizeof(inQ)); dis[Q[0]=src]=0;

    for(int f=0,r=1;f!=r;f=(f+1)%MAXV)
    {
        inQ[u=Q[f]]=false;
        for(int i=pre[u];i!=-1;i=next[i])
        {
            v=G[i].v;
            w=G[i].w;
            if(i/2!=k_th&&dis[v]>dis[u]+w)
            {
                dis[v]=dis[u]+w;

                if(!inQ[v])
                {
                    inQ[Q[r]=v]=true;
                    r=(r+1)%MAXV;
                }
            }
        }
    }

    return dis[des];
}

int tcase,n,m,u,v,w;

int main()
{
    #ifndef ONLINE_JUDGE
    freopen("Harry Potter and the Final Battle.txt","r",stdin);
    #endif

    scanf("%d",&tcase);while(tcase--)
    {
        clear();
        scanf("%d %d",&n,&m);

        int ans=-1;
        for(int i=0;m>i;i++)
        {
            scanf("%d %d %d",&u,&v,&w);
            add(u-1,v-1,w);
            add(v-1,u-1,w);
        }
        SPFA(0,n-1);
        for(int i=n-1;i!=0;i=prev[i])
        {
            ans=lmax(ans,findAns(0,n-1,curr[i]/2));
        }
        printf("%d\n",ans!=inf?ans:-1);
    }
    return 0;
}