洛谷 P1119 灾后重建（最短路_Floyd）

传送门

---

第一反应是对于每个询问建一次边跑最短路，但似乎会TLE，于是再看一眼题面
t[0] ≤ t[1] ≤ … ≤ t[N – 1]
数据保证了t是不下降的
于是我们可以边输入询问边处理，每次把t[i]<=t且没有处理过的点作为中转站跑一次Floyd，判-1，输出即可。

Code:

#include<cstdio>
#include<cstdlib>
#include<cstring>

const int INF=0x3f3f3f3f;

int a[210][210];
int t[210];
int n,m,q;

int main()
{
    memset(a,INF,sizeof(a));
    memset(t,INF,sizeof(t));
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&t[i]);
        a[i][i]=0;
    }
    for(int i=1;i<=m;i++)
    {
        int x,y,c;
        scanf("%d %d %d",&x,&y,&c);
        x++;y++;
        a[x][y]=a[y][x]=c;
    }
    scanf("%d",&q);
    int k=1;
    for(int i=1;i<=q;i++)
    {
        int sx,sy,sd;
        scanf("%d %d %d",&sx,&sy,&sd);
        sx++;sy++;
        while(t[k]<=sd)
        {
            for(int i=1;i<=n;i++)
                for(int j=1;j<=n;j++)
                    if(a[i][k]+a[k][j]<a[i][j]) a[i][j]=a[i][k]+a[k][j];
            k++;
        }
        if(a[sx][sy]>=INF || t[sx]>sd || t[sy]>sd) printf("-1\n");
        else printf("%d\n",a[sx][sy]);
    }
}