基于dijkstra实现的次短路算法
看了网上的代码，和我平常的风格不太一样，于是就自己写了一个。
dijkstra求次短路的原理是这样的：在求最短路的的过程中，我们同时更新一个dis2[]数组，表示次短路，dis2[v]有两个来源
- dis2[v]=dis2[u]+e[i].dis
- dis2[v]=dis[u]+e[i].dis
在这两个来源中找符合题意的答案即可，
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdio>
#include <queue>
#include <algorithm>
using namespace std;
const int MAXN=200005;
int init(){
int rv=0,fh=1;
char c=getchar();
while(c<'0'||c>'9'){
if(c=='-') fh=-1;
c=getchar();
}
while(c>='0'&&c<='9'){
rv=(rv<<1)+(rv<<3)+c-'0';
c=getchar();
}
return rv*fh;
}
int n,m,head[MAXN],dis[MAXN],nume,dis2[MAXN];
struct edge{
int to,nxt,dis;
}e[MAXN];
struct cmp{
bool operator () (const int &a,const int &b) const{
return dis[a]>dis[b];
}
};
void adde(int from,int to,int dis){
e[++nume].to=to;
e[nume].dis=dis;
e[nume].nxt=head[from];
head[from]=nume;
}
void sec_dij(){
priority_queue <int,vector <int> ,cmp> q;
memset(dis,0x3f,sizeof(dis));
memset(dis2,0x3f,sizeof(dis2));
q.push(1);
dis[1]=0;
while(!q.empty()){
int u=q.top();q.pop();
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].to;
if(dis[v]>dis[u]+e[i].dis){
int t=dis[v];
dis[v]=dis[u]+e[i].dis;
if(t<dis2[v]) dis2[v]=t;
q.push(v);
}else{
if(dis2[v]>dis[u]+e[i].dis&&dis[u]+e[i].dis>dis[v]){
dis2[v]=dis[u]+e[i].dis;
q.push(v);
}
}
if(dis2[v]>dis2[u]+e[i].dis&&dis2[u]+e[i].dis>dis[v]){
dis2[v]=dis2[u]+e[i].dis;
q.push(v);
}
}
}
}
int main(){
n=init();m=init();
for(int i=1;i<=m;i++){
int u=init(),v=init(),di=init();
adde(u,v,di);
adde(v,u,di);
}
sec_dij();
// cout<<dis[n]<<endl;
cout<<dis2[n]<<endl;
return 0;
}