洛谷P1119 灾后重建

#include<iostream>
using namespace std;
const int MAXN=205,INF=0x7fff;
int n,m,t[MAXN],f[MAXN][MAXN],q;

void update(int now);
int main()
{
    int now=0;
    cin>>n>>m;
    for(int i=0;i<=n;i++)
        for(int j=0;j<=n;j++)
            f[i][j]=INF;
    for(int i=0;i<n;i++)
    {
        cin>>t[i];
        f[i][i]=0;
    }
    for(int i=0;i<m;i++)
    {
        int u,v,w;
        cin>>u>>v>>w;
        f[u][v]=w;
        f[v][u]=w;
    }
    cin>>q;
    
    for(int i=1;i<=q;i++)
    {
        int u,v,d;
        cin>>u>>v>>d;
        while(t[now]<=d&&now<n)
            update(now++);
        if(f[u][v]==INF)
            cout<<"-1"<<endl;
        else if(t[u]>d||t[v]>d)
            cout<<"-1"<<endl;
        else
            cout<<f[u][v]<<endl;
    }
    return 0;
}
void update(int now)
{
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            if(f[i][j]>f[i][now]+f[j][now])
                f[i][j]=f[j][i]=f[i][now]+f[j][now];
}