描述
贝西所在的牧场一共有 N 个地点。 M 条双向通行的道路连接这些地点,其中第 i 条道路连接 Ai 和 Bi,长度为 Li。贝西想从第一个地点走到第 N 个地点,由于路上风景不错,她决定不走最短 路径,而选择次短路径。次短路径的长度严格大于最短路径。如果有两条路径的长度都是最短的,那 么它们都不算次短路径。次短路径允许重复通过一些道路或地点。请你帮助贝西找出次短路径的长度 吧,输入数据保证次短路径一定存在
输入
第一行:两个整数 N 和 M, 1 ≤ N ≤ 5000, 1 ≤ M ≤ 10^5 • 第二行到第 M + 1 行:第 i + 1 行有三个整数 Ai, Bi 和 Li, 1 ≤ Ai; Bi ≤ N; 1 ≤ Li ≤ 5000
输出
单个整数:表示从第一个点到最后一个点的次短路径长度
样例输入
4 4
1 2 100
2 4 200
2 3 250
3 4 100
样例输出
450
提示
最短路是 1 → 2 → 4,总长度为 300,次短路 是 1 → 2 → 3 → 4,总长度为 450
次短路的求法:
从起点和终点分别跑一次SPFA
然后枚举每一条边,看能不能更新次小答案就是了
#include<bits/stdc++.h>
using namespace std;
inline int read(){
char ch=getchar();
int res=0;
while(!isdigit(ch)) ch=getchar();
while(isdigit(ch)) res=(res<<3)+(res<<1)+(ch^48),ch=getchar();
return res;
}
int n,m,k,cnt,l,ans,r,last,adj[5005],nxt[200005],to[200005],val[200005],dis1[5005],dis2[5005];
bool vis[10005];
struct edge{
int u,v,w;
}e[200005];
inline void addedge(int u,int v,int w){
nxt[++cnt]=adj[u],adj[u]=cnt,to[cnt]=v,val[cnt]=w;
nxt[++cnt]=adj[v],adj[v]=cnt,to[cnt]=u,val[cnt]=w;
}
inline void spfa(int s,int *dis){
int q[500005],qn;
memset(vis,false,sizeof(vis));
last=1e9;
q[qn=1]=s,vis[s]=true,dis[s]=0;
for(int ql=1;ql<=qn;ql++){
int u=q[ql];
for(int e=adj[u];e;e=nxt[e]){
int v=to[e];
if(dis[u]+val[e]<dis[v])
{
dis[v]=dis[u]+val[e];
if(!vis[v]){
vis[v]=true;
q[++qn]=v;
}
}
}
vis[u]=false;
}
return;
}
int main(){
n=read(),m=read();
for(int i=1;i<=m;i++){
int u=read(),v=read(),w=read();
e[i].u=u,e[i].v=v,e[i].w=w;
addedge(u,v,w);
}
memset(dis1,127/3,sizeof(dis1));
memset(dis2,127/3,sizeof(dis2));
spfa(1,dis1);
spfa(n,dis2);
int ans=1e9;
for(int i=1;i<=m;i++){
int u=e[i].u,v=e[i].v;
int tmp=dis1[u]+dis2[v]+e[i].w;
if(tmp>dis1[n]&&tmp<ans){
ans=tmp;
}
tmp=dis1[v]+dis2[u]+e[i].w;
if(tmp>dis1[n]&&tmp<ans){
ans=tmp;
}
}
cout<<ans<<'
';
}