【图论】SPFA算法求第二最短路经

题目 - 第二最短路

描述

在以前的一些课程上，我们已经学习过最短路的几个经典算法（Dijkstra, Floyd, Bellman-  Ford)，现在，不妨来考虑一下一个稍微复杂一点点的问题：第二最短路。  给出一个n个点，m条边的无向图，请你找出从点1到点n的第二最短路：这条通路的长度比最短路要长，但是除最短路外的所有通路的长度都不比这条通路更短（可能相同，即第二最短路可能有很多条）。  第二最短路中允许有重复的点和边

关于输入

第一行是两个整数n和m(1<=n<=5000,1<=m<=100000)。 接下来m行，每行三个整数a,b,c，表示点a与点b之间有一条长为c的边。

关于输出

输出第二最短路的长度

例子输入

4 4 1 2 100 2 4 200 2 3 250 3 4 100

例子输出

450

提示

由于是无向图，第二最短路有两种情况：  1. 在最短路的基础上加一个小环（怎么加？）  2. 对于某条普通边(v, u)，从1到v的最短路+(v,u)+从u到n的最短路（也就是n到u）。  时间复杂度与单源最短路算法一致。

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<queue>
using namespace std;
struct Road{
int x,y,wei;
bool operator<(const Road& r)const{
if(x==r.x){
if(y==r.y) return wei<r.wei;
return y<r.y;
}
return x<r.x;
}
};
Road road[200005];
queue<int> q;
int n,m,pos[5005],dis1_n[5005],disn_1[5005],dis[5005];


void spfa(int begin);
int main(){
int minRoad,result;
scanf("%d%d",&n,&m);
for(int i=0;i<=n;i++)
pos[i]=0;
for(int i=0;i<2*m;i++){
scanf("%d%d%d",&road[i].x,&road[i].y,&road[i].wei);
i++;
road[i].x=road[i-1].y;
road[i].y=road[i-1].x;
road[i].wei=road[i-1].wei;
pos[road[i].x]++;
pos[road[i].y]++;
}
for(int i=2;i<=n;i++)
pos[i]+=pos[i-1];
sort(road,road+2*m);
spfa(1);
for(int i=1;i<=n;i++)
dis1_n[i]=dis[i];
spfa(n);
for(int i=1;i<=n;i++)
disn_1[i]=dis[i];
minRoad=disn_1[1];
//cout<<"minRoad: "<<minRoad<<endl;
result=999999;
for(int i=0;i<2*m;i++){
result=min(result,minRoad+2*road[i].wei);
if(dis1_n[road[i].x]+disn_1[road[i].y]+road[i].wei<=minRoad)
continue;
result=min(result,dis1_n[road[i].x]+disn_1[road[i].y]+road[i].wei);
}
cout<<result<<endl;
//system("pause");
return 0;
}


void spfa(int begin){
int t;
while(q.empty()==false)
q.pop();
for(int i=1;i<=n;i++){
dis[i]=999999;
}
q.push(begin);
dis[begin]=0;
while(q.empty()==false){
int tmp=q.front();
q.pop();
for(int i=pos[tmp]-1;road[i].x==tmp;i--){
//cout<<"haha"<<endl;
t=road[i].y;
if(dis[tmp]+road[i].wei<dis[t]){
dis[t]=dis[tmp]+road[i].wei;
q.push(t);
}
}
}
return;
}