hdu 3986 Harry Potter and the Final Battle 最短路spfa+枚举

题目链接

题意：给出n个点，m条边，且有重边。求除去任意一条边的最长最短路。

由于有重边的情况，所以用邻接表来存储。注意标记毁坏的边时，正反双向都要标记。

#include <iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
#define N 1100
#define INF 0x7ffffff
#define M 110000

using namespace std;

struct node
{
    int to,next,w;
}e[M];

int cnt,head[N],ds,d[N],v[N],n,m,pre[N],path[N];

void add_edge(int u,int v,int w)
{
    e[cnt].to=v;
    e[cnt].w=w;
    e[cnt].next=head[u];
    head[u]=cnt++;
}

void spfa(int t)
{
    for(int i=1;i<=n;i++)   d[i]=INF,v[i]=0;
    queue<int> q;
    q.push(1);
    v[1]=1;
    d[1]=0;
    while(!q.empty())
    {
        int c=q.front();
        q.pop();
        v[c]=0;
        for(int i=head[c];i!=-1;i=e[i].next)
        {
            int u=e[i].to;
            if(i!=(ds^1)&&i!=ds&&d[u]>d[c]+e[i].w)
            {
                d[u]=d[c]+e[i].w;
                if(t) pre[u]=c,path[u]=i;
                if(!v[u])   v[u]=1,q.push(u);
            }
        }
    }
}

int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        cin>>n>>m;
        memset(head,-1,sizeof(head));
        cnt=0;
        for(int i=0;i<m;i++)
        {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            add_edge(u,v,w);
            add_edge(v,u,w);
        }
        memset(pre,-1,sizeof(pre));
        ds=-1;
        spfa(1);
        if(d[n]==INF)
        {
            cout<<-1<<endl;
            continue;
        }
        int ans=-1;
        for(int i=n;i!=1;i=pre[i])
        {
            ds=path[i];
            spfa(0);
            ans=max(ans,d[n]);
        }
        if(ans>=INF)    cout<<-1<<endl;
        else    cout<<ans<<endl;
    }
}