算法提高 道路和航路 【SPFA+SLF优化】

http://lx.lanqiao.cn/problem.page?gpid=T22

#include <bits/stdc++.h>
using namespace std;

const int maxn=150005;
const int INF=1<<30;

struct Edge{
    int to,next,val;
}e[maxn];

int N,M1,M2,S;
int dis[maxn];
int head[maxn];
int num=0;
bool vis[maxn];

void init(){
    for (int i=0;i<maxn;i++){
        dis[i]=INF;
    }
    memset(vis,false,sizeof(vis));
    memset(head,-1,sizeof(head));
}

void add(int from,int to,int val){
    e[num].to=to;
    e[num].val=val;
    e[num].next=head[from];
    head[from]=num++;
}

void SPFA(int s){

    deque <int> q;
    dis[s]=0;
    vis[s]=1;
    q.push_back(s);

    while (!q.empty()){
        int u=q.front();
        q.pop_front();
        vis[u]=false;
        for (int i=head[u];i!=-1;i=e[i].next){
            int v=e[i].to;
            if (dis[v]>dis[u]+e[i].val){
                dis[v]=dis[u]+e[i].val;
                if (!vis[v]){
                    vis[v]=1;
                    //SLF优化
                    if (q.empty() || dis[v]<dis[q.front()]){
                        q.push_front(v);
                    }
                    else{
                        q.push_back(v);
                    }
                }
            }
        }
    }
}

int main(){

    init();

    scanf("%d%d%d%d",&N,&M1,&M2,&S);
    int u,v,w;
    for (int i=0;i<M1;i++){
        scanf("%d%d%d",&u,&v,&w);
        add(u,v,w);
        add(v,u,w);
    }
    for (int i=0;i<M2;i++){
        scanf("%d%d%d",&u,&v,&w);
        add(u,v,w);
    }

    SPFA(S);

    for (int i=1;i<=N;i++){
        if (dis[i]==INF){
            printf("NO PATH\n");
        }
        else{
            printf("%d\n",dis[i]);
        }
    }
}