【洛谷】P1629 邮递员送信

题目地址：

https://www.luogu.com.cn/problem/P1629

题目描述：
有一个邮递员要送东西，邮局在节点$1$。他总共要送$n-1$样东西，其目的地分别是节点$2$到节点$n$。由于这个城市的交通比较繁忙，因此所有的道路都是单行的，共有$m$条道路。这个邮递员每次只能带一样东西，并且运送每件物品过后必须返回邮局。求送完这$n-1$样东西并且最终回到邮局最少需要的时间。

输入格式：
第一行包括两个整数，$n$和$m$，表示城市的节点数量和道路数量。
第二行到第$(m+1)$行，每行三个整数，$u,v,w$，表示从$u$到$v$有一条通过时间为$w$的道路。

输出格式：
输出仅一行，包含一个整数，为最少需要的时间。

数据范围：
对于$30\%$的数据，$1\le n\le 200$。
对于$100\%$的数据，$1\le n\le 10^3$，$1\le m\le 10^5$，$1\le u,v\le n$，$1\le w\le 10^4$，输入保证任意两点都能互相到达。

送邮件的时候可以视为求以$1$为源点的单源最短路问题，而返回邮局的时候可以视为求以$1$为源点，在图中每条边都反向的新图中求单源最短路问题。即两次Dijkstra算法即可。建图的时候，可以同时建出反向图。代码如下：

#include <iostream>
#include <cstring>
#include <queue>
using namespace std;
using PII = pair<int, int>;

const int N = 1010, M = 2e5 + 10;
int n, m;
int h[N * 2], e[M], ne[M], w[M], idx;
int dist[N];
bool vis[N];
long res;

void add(int a, int b, int c) {
    e[idx] = b, ne[idx] = h[a], w[idx] = c, h[a] = idx++;
}

// d是偏移量
void dijkstra(int d) {
    memset(vis, 0, sizeof vis);
    memset(dist, 0x3f, sizeof dist);
    dist[1] = 0;
    priority_queue<PII, vector<PII>, greater<>> heap;
    heap.push({0, 1 + d});
    while (heap.size()) {
        auto t = heap.top();
        heap.pop();
        int v = t.second;
        if (vis[v - d]) continue;
        vis[v - d] = true;
        for (int i = h[v]; ~i; i = ne[i]) {
            int j = e[i];
            if (!vis[j - d] && dist[j - d] > dist[v - d] + w[i]) {
                dist[j - d] = dist[v - d] + w[i];
                heap.push({dist[j - d], j});
            }
        }
    }
}

int main() {
    scanf("%d%d", &n, &m);
    memset(h, -1, sizeof h);
    for (int i = 1; i <= m; i++) {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        add(a, b, c), add(b + N, a + N, c);
    }

    dijkstra(0);
    for (int i = 2; i <= n; i++) res += dist[i];

    dijkstra(N);
    for (int i = 2; i <= n; i++) res += dist[i];

    printf("%ld\n", res);

    return 0;
}

时间复杂度$O(m\log n)$，空间$O(n)$。