1. 程式人生 > >SPFA最短路徑 路徑記錄

SPFA最短路徑 路徑記錄

做了一道板子題,求v0點到任意點的最短距離和最短路徑

這裡用SPFA,記錄一下

#include <bits/stdc++.h>
#define MAXN 500005

using namespace std;
typedef long long ll;

struct Edge{
    ll next, weight;
    
    Edge(ll _next, ll _weight){
        next = _next;
        weight = _weight;
    }
};

vector<Edge> g[MAXN];//鄰接表

ll n, m;
ll dis[MAXN], vis[MAXN];
ll pre[MAXN];

void spfa(ll s){
    queue<ll> q;
    for(ll i = 0; i <= MAXN; i++)
        dis[i] = MAXN << 4;
    memset(vis, 0, sizeof vis);
    q.push(s);
    vis[s] = 1;
    dis[s] = 0;
    while(!q.empty()){
        ll u = q.front();
        q.pop();
        vis[u] = 0;
        for(ll i = 0; i < g[u].size(); i++){
            ll v = g[u][i].next;
            if(dis[v] > dis[u] + g[u][i].weight){
                dis[v] = dis[u] + g[u][i].weight;
                pre[v] = u;//記錄前驅
                if(!vis[v]){
                    vis[v] = 1;
                    q.push(v);
                }
            }
        }
    }
}

int main(){
    scanf("%lld %lld", &n, &m);
    ll a, b, v;
    for(ll i = 0; i < m; i++){
        scanf("%lld %lld %lld", &a, &b, &v);
        g[a].push_back(Edge(b, v));
        g[b].push_back(Edge(a, v));//無向圖雙向儲存
    }
    for(ll i = 1; i < n; i++){
        printf("v0 to v%lld\n", i);
        memset(pre, 0, sizeof pre);
        spfa(i);//從終點反向跑spfa
        printf("Length: %lld\n", dis[0]);
        printf("Path: \n");
        ll j = 0;
        printf("0");
        while(j != i){
            printf("->%lld",pre[j]);
            j=pre[j];
        }
        printf("\n\n");
    }
    return 0;
}