【ACWing】1132. 农场派对

题目地址：

https://www.acwing.com/problem/content/1134/

$N$头牛要去参加在某农场举行的一场编号为$X$的牛的派对。有$M$条有向道路，每条路长$T_i$；每头牛参加完派对后都必须回到家，每头牛都会选择最短路径。求这$N$头牛的最短路径（一个来回）中最长的一条的长度。特别提醒：可能有权值不同的重边。

输入格式：
第一行包含三个整数$N,M,X$。接下来$M$行，每行包含三个整数$A_i,B_i,T_i$，表示有一条从$A_i$到$B_i$的路，长度为$T_i$。

输出格式：
共一行，一个数，表示最短路中最长的一条的长度。

数据范围：
$1\le N\le 1000$
$1\le X\le N$
$1\le M\le 10^5$
$1\le T_i\le 100$

可以建正向图和反向图来两遍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, X;
int h[2][N], e[M], ne[M], w[M], idx;
int dist[2][N];
bool vis[N];

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

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

int main() {
    scanf("%d%d%d", &n, &m, &X);
    memset(h, -1, sizeof h);
    while (m--) {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        add(h[0], a, b, c), add(h[1], b, a, c);
    }

    memset(dist, 0x3f, sizeof dist);
    dijkstra(0, 0);
    dijkstra(1, 1);

    int res = 0;
    for (int i = 1; i <= n; i++)
        res = max(res, dist[0][i] + dist[1][i]);

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

    return 0;
}

时间复杂度$O(M\log N)$，空间$O(N)$。