【Gym - 100812G 】Short Path （SPFA）

BUPT2017 wintertraining(15) #7B

题意

n个点m条无向有权边(2 ≤ n ≤ 10^5, 1 ≤ m ≤ 10^5)，每个点标记了0或1，求所有1中，最近的两个1的下标及距离。

题解

先用SPFA求出每个点离标记1的点最近的距离，d[i]。
同时记录下每个点最近的1的下标。
两个最近的1，要么是被一条边连着，要么是被几个0隔着的边连着。
我们通过寻找它们中间的边来找出它们。
枚举每条边，如果相邻都是1，或者都是0且最近的1不是同一个，或者一个1，一个0，那么这条边两个端点的最近1的距离，就可以拿来更新答案。
需要long long。

代码

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <vector>
#include <queue>
#define inf 0x3f3f3f3f3f3f3f3f
#define N 100005
#define ll long long
using namespace std;

typedef pair<ll,int> PII;
vector<PII> edge[N<<1];

int n,m,a[N];
ll d[N];
int near[N];
bool b[N];
queue <int> q;

void SPFA(){
    for(int i=1;i<=n;++i)if(!d[i]) {
        q.push(i);
        b[i]=true;
        near[i]=i;
    }
    while(!q.empty()){
        int u=q.front(); q.pop(); b[u]=false;
        for(int j=0;j<edge[u].size();++j){
            int v = edge[u][j].second;
            ll w = edge[u][j].first;
            if(d[u]+w<d[v]){
                d[v]=d[u]+w;
                near[v]=near[u];
                if(!b[v]){
                    q.push(v);
                    b[v]=true;
                }
            }
        }
    }
}
int main() {
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;++i){
        scanf("%d",&a[i]);
        d[i]=a[i]?0:inf;
    }
    for(int i=1;i<=m;++i){
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        edge[u].push_back(make_pair(w,v));
        edge[v].push_back(make_pair(w,u));
    }
    SPFA();
    int ansa,ansb;
    ll ans=inf;
    for(int i=1;i<=n;++i)
    for(int j=0;j<edge[i].size();++j){
        int to = edge[i][j].second;
        ll w = edge[i][j].first;
        if(a[i]&&a[to]){
            if(ans>w){
                ans=w;
                ansa=i;
                ansb=to;
            }
        }else if(near[i]!=near[to]){
            if(ans>d[i]+d[to]+w){
                ans=d[i]+d[to]+w;
                ansa=near[i];
                ansb=near[to];
            }
        }
    }
    if(ans!=inf)printf("%lld\n%d %d",ans,ansa,ansb);
    else puts("No luck at all");
    return 0;
}