次短路模板

基于dijkstra实现的次短路算法

看了网上的代码，和我平常的风格不太一样，于是就自己写了一个。
dijkstra求次短路的原理是这样的：在求最短路的的过程中，我们同时更新一个dis2[]数组，表示次短路，dis2[v]有两个来源

1. dis2[v]=dis2[u]+e[i].dis
2. 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;
}