Codevs1021题解---SPFA+路径记录

题目描述 Description

麦克找了个新女朋友，玛丽卡对他非常恼火并伺机报复。

因为她和他们不住在同一个城市，因此她开始准备她的长途旅行。

在这个国家中每两个城市之间最多只有一条路相通，并且我们知道从一个城市到另一个城市路上所需花费的时间。

麦克在车中无意中听到有一条路正在维修，并且那儿正堵车，但没听清楚到底是哪一条路。无论哪一条路正在维修，从玛丽卡所在的城市都能到达麦克所在的城市。

玛丽卡将只从不堵车的路上通过，并且她将按最短路线行车。麦克希望知道在最糟糕的情况下玛丽卡到达他所在的城市需要多长时间，这样他就能保证他的女朋友离开该城市足够远。

编写程序，帮助麦克找出玛丽卡按最短路线通过不堵车道路到达他所在城市所需的最长时间(用分钟表示)。

输入描述 Input Description

第一行有两个用空格隔开的数N和M，分别表示城市的数量以及城市间道路的数量。1≤N≤1000，1≤M≤N*(N-1)/2。城市用数字1至N标识，麦克在城市1中，玛丽卡在城市N中。

接下来的M行中每行包含三个用空格隔开的数A，B和V。其中1≤A，B≤N，1≤V≤1000。这些数字表示在A和城市B中间有一条双行道，并且在V分钟内是就能通过。

输出描述 Output Description

输出文件的第一行中写出用分钟表示的最长时间，在这段时间中，无论哪条路在堵车，玛丽卡应该能够到达麦克处，如果少于这个时间的话，则必定存在一条路，该条路一旦堵车，玛丽卡就不能够赶到麦克处。

样例输入 Sample Input

5 7

1 2 8

1 4 10

2 3 9

2 4 10

2 5 1

3 4 7

3 5 10

样例输出 Sample Output

27

这道题乍一看是个单源最短路，只有1000的数据几乎是想咋做就咋做。

但再仔细一看输出描述，才发现要处理堵车的情况，想出来的解法很暴力，先SPFA一遍并记录路径，再依次删掉路径中的每条边再SPFA，取最大值。

那么如何记录路径呢？

我的方法是用一个pre数组来记录到当前节点的最优点/边。

1，pre[i]表示从1到i的最短路上的上一个节点，如图：

pre[i]表示1到点i的最短路的前一个点的编号，只需不停地x=pre[x]再找到对应的边删除即可。

2，pre[i]表示从1到i的最短路上一条边的编号，如图：

pre[i]表示到1点i的最短路的前一条边的编号，int edge=pre[x];再循环edge=pre[e[edge].u];只需将编号为edge的边删除。

本题是无向边，需要再删除另一个方向上的边，但在邻接表存边时一条无向边被拆为两条有向边，且编号相邻，所以找到一条边时再删去+1或-1的边。

在SPFA的过程中，每当松弛一条边时边更新pre数组，即可记录下路径。

代码如下，使用pre数组记录边的编号。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define inf 100000000
int d[1001],q[100001],head[1001],pre[1001],n,m,cnt,h,t,ans;
bool vis[10001];
struct edge
{
    int u,v,val,next;
    bool no;//是否堵车
}e[5000050];
void spfa(bool re)//re为true则记录路径
{
    h=t=1;
    q[t++]=1;
    while(h<t)
    {
        for(int i=head[q[h]];i!=0;i=e[i].next)
        {
            if(e[i].no)continue;
            if(d[q[h]]+e[i].val<d[e[i].v])
            {
                d[e[i].v]=d[q[h]]+e[i].val;
                if(re)pre[e[i].v]=i;//记录边的编号i
                if(!vis[e[i].v])
                {
                    q[t++]=e[i].v;
                    vis[e[i].v]=true;
                }
            }
        }
        vis[q[h]]=false;
        ++h;
    }
}
void add(int uu,int vv,int ww)//一条无向边被拆为二条有向边储存，编号分别为cnt,cnt+1
{
    e[++cnt].next=head[uu];
    head[uu]=cnt;
    e[cnt].u=uu;
    e[cnt].v=vv;
    e[cnt].val=ww;
    e[++cnt].next=head[vv];
    head[vv]=cnt;
    e[cnt].u=vv;
    e[cnt].v=uu;
    e[cnt].val=ww;
}
int main()
{
    int x,y,w,now;
    scanf("%d%d",&n,&m);
    for(int i=2;i<=n;i++)
        d[i]=inf;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&x,&y,&w);
        add(x,y,w);
    }
    spfa(true);
    ans=d[n];
    now=pre[n];
    while(1)
    {
        if(now+1<=cnt&&e[now+1].u==e[now].v&&e[now+1].v==e[now].u)//找到另一条应删去的边
        {
            e[now+1].no=e[now].no=true;
            memset(vis,false,sizeof(vis));
            for(int i=2;i<=n;i++)d[i]=inf;
            spfa(false);
            e[now+1].no=e[now].no=false;
            ans=max(ans,d[n]);
        }
        else if(now-1>=1&&e[now-1].u==e[now].v&&e[now-1].v==e[now].u)
        {
            e[now-1].no=e[now].no=true;
            memset(vis,false,sizeof(vis));
            for(int i=2;i<=n;i++)d[i]=inf;
            spfa(false);
            e[now-1].no=e[now].no=false;
            ans=max(ans,d[n]);
        }
        if(e[now].u==1)break;//若边的起点为1，则退出
        now=pre[e[now].u];//下一条边
    }
    printf("%d",ans);
    return 0;
}