洛谷P1629 郵遞員送信【最短路】

題目連結：P1629 郵遞員送信

程式說明：

注意每次只能拿一件物品，送到目的地後必須返回起點才能送下一件。

因此可以將題目轉化為：從1號點到其餘各點的最短路，再加上其餘各點到1號點的最短路即為正確答案。

求多個點到1號點的最短路，可以將鄰接矩陣座標互換（即將有向圖反轉）。

程式碼如下:

#include <iostream>
#include <cstring>
using namespace std;
const int N = 1010, M = 100010;
int g[N][N], dist[N], st[N], n, m, res;

void dijkstra() {
    memset(dist, 0x3f, sizeof dist);
    memset(st, 0, sizeof st);
    dist[1] = 0;

    for(int i = 0; i < n; i++) {
        int t = -1;
        for(int j = 1; j <= n; j++) 
            if(!st[j] && (t == -1 || dist[j] < dist[t]))
                t = j;

        st[t] = 1;
        for(int j = 1; j <= n; j++)
            dist[j] = min(dist[j], dist[t] + g[t][j]);
    }
}
int main() {
    memset(g, 0x3f, sizeof g);
    cin>>n>>m;
    while(m--) {
        int x, y, z;
        cin>>x>>y>>z;
        g[x][y] = min(g[x][y], z);
    }       
    dijkstra();
    for(int i = 1; i <= n; i++)
        res += dist[i];
    //將圖反轉
    for(int i = 1; i <= n; i++)
        for(int j = i + 1; j <= n; j++)
            swap(g[i][j], g[j][i]);
     
    dijkstra();
    for(int i = 1; i <= n; i++)
        res += dist[i];
    cout<<res;
    return 0;
}