洛谷 P1119 灾后重建 理解Floyed算法

洛谷 P1119 灾后重建 理解Floyed算法

题解：

要理解Floyed算法的原理，每次取前k个点作为中转，不断更新。
不能硬背模板。

代码如下：

#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<cmath>
#include<queue>
#include<cstring>
#include<vector>
#define MAX 205
#define INF 0x3f3f3f3f
typedef long long ll;
using namespace std;

int n,m,q,s,e,t,a[MAX],f[MAX][MAX];//a[i]记录第i个村庄重建完成的时间
                           //f[i][j]表示从i到j的长度

void floyed(int k){
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
             if(!(i==j||i==k||j==k)){//i不能等于j， j不能等于k， i不能等于k
                    f[i][j]=min(f[i][k]+f[k][j],f[i][j]);//取最小值
             }
        }
    }
}

int main(){
    memset(f, INF, sizeof(f));//初始化f
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++){//读入每个村庄重建好的时间
        scanf("%d",&a[i]);
    }
    int b,c,len;
    for(int i=0;i<m;i++){
        scanf("%d%d%d",&b,&c,&len);
        f[b][c]=f[c][b]=len;
    }
    scanf("%d",&q);//读入询问的次数
    int now=0;
    for(int i=0;i<q;i++){
        scanf("%d%d%d",&s,&e,&t);
        while(a[now]<=t&&now<n){
            floyed(now);
            now++;
        }
        if(a[s]>t||a[e]>t){//说明起点和终点的城市有没有建好的情况
            printf("-1\n");
        }else{
            if(f[s][e]==INF){//说明两点不连通
                printf("-1\n");
            }else{//否则输出最短路径
                printf("%d\n",f[s][e]);
            }
        }
    }
    return 0;
}