多源最短路-Floyd

#include <iostream>
#include <queue>
using namespace std;

typedef long long ll;
const int MAXSIZE=210;
const int INF=(int)1e9;

int matrix[MAXSIZE][MAXSIZE];
int ans[MAXSIZE][MAXSIZE][MAXSIZE];
int n,m,t[MAXSIZE];

void Floyd()
{
    for (int k=0;k<n;k++)
    {
        for (int i=0;i<n;i++)
        {
            for (int j=0;j<n;j++)
            {
                matrix[i][j]=min(matrix[i][j],matrix[i][k]+matrix[k][j]);
            }
        }

        for (int i=0;i<n;i++)
        {
            for (int j=0;j<n;j++)
            {
                ans[k][i][j]=matrix[i][j];
            }
        }
    }
    return;
}

void query(int x,int y,int d)
{
    if (t[x]>d || t[y]>d)   //起点或终点村庄还未重建完成
    {
        cout<<"-1"<<endl;
        return;
    }
    else
    {
        int ret=INF;
        for (int k=0;k<n;k++)   //遍历x至y的每个中间村庄
        {
            if (t[k]<=d && ans[k][x][y]<INF)   //中间村庄已重建并且能走通
            {
                ret=min(ret,ans[k][x][y]);
            }
        }
        if (ret!=INF)   cout<<ret<<endl;
        else    cout<<-1<<endl;
    }
    return;
}

int main()
{
    cin>>n>>m;
    for (int i=0;i<n;i++)
    {
        cin>>t[i];
    }

    for (int i=0;i<n;i++)
    {
        for (int j=0;j<n;j++)
        {
            if (i!=j)   matrix[i][j]=INF;
        }
    }

    int u,v,w;
    for (int i=0;i<m;i++)
    {
        cin>>u>>v>>w;
        matrix[u][v]=matrix[v][u]=w;
    }

    Floyd();

    int q,x,y,d;
    cin>>q;
    while (q--)
    {
        cin>>x>>y>>d;
        query(x,y,d);
    }

    return 0;
}