灾后重建(floyd更新中转点)

P1119 灾后重建 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

题目描述

给出 B 地区的村庄数 N，村庄编号从 0 到 N−1，和所有 M 条公路的长度，公路是双向的。并给出第 i 个村庄重建完成的时间 ti​，你可以认为是同时开始重建并在第 ti​ 天重建完成，并且在当天即可通车。若 ti​ 为 0 则说明地震未对此地区造成损坏，一开始就可以通车。之后有 Q 个询问 (x,y,t)，对于每个询问你要回答在第 t 天，从村庄 x 到村庄 y 的最短路径长度为多少。如果无法找到从 x 村庄到 y 村庄的路径，经过若干个已重建完成的村庄，或者村庄 x 或村庄 y 在第 t 天仍未重建完成，则需要返回 -1。

思路：

g[i][j]  = min(g[i][j], g[i][k] + g[k][j]); 代表只经过前k个点的中转更新 g[i][j],因为题目保证了每次询问给出的时间使递增的，所以每次询问都更新到第k个点即可

#include <algorithm>
#include <cstring>
#include <iostream>
#define dbg(x) cout << #x << '=' << x << endl
#define FA ios::sync_with_stdio(0), cin.tie(0), cout.tie(0)
#define INF 0x3f3f3f3f
using namespace std;
const int N = 210, M = 1e3 + 10, mod = 1e9 + 7;
const double eps = 1e-8;
typedef long long LL;
typedef pair<int, int> PII;
int n, m;
int g[N][N];
int t[N];

void floyd(int k) { //代表只经过前k个点更新出来的最短路
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            g[i][j] = g[j][i] = min(g[i][j], g[i][k] + g[k][j]);
        }
    }
}

void solve() {
    scanf("%d %d", &n, &m);
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            if (i == j)
                g[i][j] = 0;
            else
                g[i][j] = INF;
        }
    }
    for (int i = 0; i < n; i++) {
        scanf("%d", &t[i]);
    }
    while (m--) {
        int a, b, c;
        scanf("%d %d %d", &a, &b, &c);
        g[a][b] = g[b][a] = min(g[a][b], c);  //无向图建边判断重边
    }
    int q;
    scanf("%d", &q);
    int now = 0;

    while (q--) {
        int a, b, c;
        scanf("%d %d %d", &a, &b, &c);
        while (t[now] <= c && now <= n - 1) { //更新条件
            floyd(now);
            now++;
        }
        if (t[a] > c || t[b] > c) {   //注意判断两个村庄是否以及建成
            printf("-1\n");
            continue;
        }
        if (g[a][b] < INF / 2)
            printf("%d\n", g[a][b]);
        else
            printf("-1\n");
    }
}
signed main() {
    int T = 1;
    // FA;
    while (T--) {
        solve();
    }
}