次短路模板

Posted 2020-10-19 Mr_Wolfram的高维空间

基于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;
}