洛谷 P1119 灾后重建

题目链接：https://www.luogu.org/problem/P1119

借用大佬的一句话：floyd的算法本质，

从i城市到j城市，通过前k个城市的贪心后得到的距离矩阵，即得到的距离矩阵是通过前K个城市（的贪心）之后的最短路。

那么这题就变得简单了。

每个城市重建时间一定是非递减的，之后访问的时间也是非递减的，那么我们可以用之后的访问时间来确定，能经过哪K个城市，

那么如果输入的天数不能从一个城市到达那个城市的最短路判断条件就是判断

f[u][v] == inf  || tim[u] > t || tim[v] > t  。

---

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <stack>
#include <set>
#include <cmath>
#include <string>
#include <map>
#include <cmath>
#include <iomanip>
using namespace std;
 
typedef long long LL;
#define inf 1e8
#define rep(i,j,k) for(int i = (j); i <= (k); i++)
#define rep__(i,j,k) for(int i = (j); i < (k); i++)
#define per(i,j,k) for(int i = (j); i >= (k); i--)
#define per__(i,j,k) for(int i = (j); i > (k); i--)

const int N = (int)2e2 + 10;
int f[N][N];
int tim[N];

int main(){ 

    int n,m;
    scanf("%d%d",&n,&m);

    rep__(i,0,n) scanf("%d",tim + i);
    rep__(i,0,n) rep__(j,0,n){
        if(i == j) f[i][j] = 0;
        else f[i][j] = inf;
    }
    int u,v,w,t;
    rep__(i,0,m){
        scanf("%d%d%d",&u,&v,&w);
        f[u][v] = f[v][u] = w;
    }
    int q;
    scanf("%d",&q);
    int k = 0;
    while(q--){
        scanf("%d%d%d",&u,&v,&t);
        while(k < n && tim[k] <= t){
            rep(i,0,n - 1) rep(j,0,n - 1){
                f[i][j] = min(f[i][j], f[i][k] + f[k][j]);
            }
            k++;
        }

        if(f[u][v] == inf  || tim[u] > t || tim[v] > t) printf("-1\n");
        else printf("%d\n",f[u][v]);
    }

    getchar(); getchar();
    return 0;
}