题意:某道路共有R条道路,N个路口。问从1号路口到N路口的次短路是?次短路是比最短路长度次短的路径。
思路:Dijkstra 算法虽然只能求最短路,但是我们修改一下。次短路:要么是到终点的前某一个点的最短路加上它到终点的路,或者是要么到它的次短路加上到终点的最短路。因此,对于每个顶点,我们记录的不仅是最短路,还有 次短路距离。接下来就是和Dijkstra()一样的做法,不断更新这两个距离就可以求出次短路。
Dijkstra用的是优先队列优化过的,挑战代码真精炼,可以直接当模板用了。
#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
const int N=5050;
typedef pair<int,int> P;
struct edge{
int to,w;
};
vector<edge>G[N];
int dist[N],dist2[N],n,m;
const int inf=2<<29;
void solve()
{
priority_queue<P,vector<P>,greater<P> >que;
for(int i=1;i<=n;i++) dist[i]=inf;
for(int i=1;i<=n;i++) dist2[i]=inf;
dist[1]=0;
que.push(P(0,1));
while(!que.empty()){
P now=que.top();
que.pop();
int v=now.second,d=now.first;
if(dist2[v]<d) continue;
for(int i=0;i<G[v].size();i++){
edge &e=G[v][i];
int d2=d+e.w;
if(dist[e.to]>d2){
swap(dist[e.to],d2);
que.push(P(dist[e.to],e.to));
}
if(dist2[e.to]>d2){
swap(dist2[e.to],d2);
que.push(P(dist2[e.to],e.to));
}
}
}
printf("%d\n",dist2[n]);
}
int main()
{
while(~scanf("%d%d",&n,&m)){
for(int i=0;i<m;i++){
int a,b,c;scanf("%d%d%d",&a,&b,&c);
G[a].push_back(edge{b,c});
G[b].push_back(edge{a,c});
}
solve();
}
}