P1119 灾后重建

P1119 灾后重建

题目背景

B地区在地震过后，所有村庄都造成了一定的损毁，而这场地震却没对公路造成什么影响。但是在村庄重建好之前，所有与未重建完成的村庄的公路均无法通车。换句话说，只有连接着两个重建完成的村庄的公路才能通车，只能到达重建完成的村庄。
题目描述

给出B地区的村庄数N，村庄编号从000到N−1，和所有MMM条公路的长度，公路是双向的。并给出第i个村庄重建完成的时间ti，你可以认为是同时开始重建并在第ti天重建完成，并且在当天即可通车。若ti为000则说明地震未对此地区造成损坏，一开始就可以通车。之后有Q个询问(x,y,t)，对于每个询问你要回答在第t天，从村庄x到村庄y的最短路径长度为多少。如果无法找到从x村庄到y村庄的路径，经过若干个已重建完成的村庄，或者村庄x或村庄y在第t天仍未重建完成 ，则需要返回−1。

输入格式

第一行包含两个正整数N,M，表示了村庄的数目与公路的数量。

第二行包含NNN个非负整数t0,t1,…,tN−1​，表示了每个村庄重建完成的时间，数据保证了t0≤t1≤…≤tN−1。

接下来M行，每行3个非负整数i,j,w为不超过10000的正整数，表示了有一条连接村庄i与村庄jjj的道路，长度为w，保证i≠ji≠j，且对于任意一对村庄只会存在一条道路。

接下来一行也就是M+3行包含一个正整数Q，表示Q个询问。

接下来Q行，每行333个非负整数x,y,tx，询问在第t天，从村庄x到村庄y的最短路径长度为多少，数据保证了t是不下降的。

输出格式

共Q行，对每一个询问(x,y,t)输出对应的答案，即在第ttt天，从村庄x到村庄y的最短路径长度为多少。如果在第t天无法找到从x村庄到y村庄的路径，经过若干个已重建完成的村庄，或者村庄x或村庄y在第t天仍未修复完成，则输出−1。

输入输出样例

输入

4 5
1 2 3 4
0 2 1
2 3 1
3 1 2
2 1 4
0 3 5
4
2 0 2
0 1 2
0 1 3
0 1 4

输出

-1
-1
5
4

这里要注意询问所给出的时间是递增的，每个村庄所用时间也是递增的，这就可以给本题进行有个小小的优化，让原来Floyed算法O(QN^3)优化到O(N^3+Q)，不然会超时，具体操作就是对中转点k进行优化，直接看代码里的注释

#include <bits/stdc++.h>

using namespace std;

const int maxn = 1005;
const int inf = 0x3f3f3f3f;
int t[maxn];
int f[maxn][maxn];
int n,m,q;
int x,y,tt;


int main()
{
    fill(f[0],f[0]+maxn*maxn,inf);
    fill(t,t+maxn,inf);
    scanf("%d %d",&n,&m);
    for(int i = 0 ; i < n ; i++)
        scanf("%d",&t[i]);
    for(int i = 0 ; i < m ; i++)
    {
        int u,v,w;
        scanf("%d %d %d",&u,&v,&w);
        f[u][v] = f[v][u] = w;
    }
    scanf("%d",&q);
    int k = 0;
    while(q--)
    {
        scanf("%d %d %d",&x,&y,&tt);
        while(t[k] <= tt)//由于询问是递增的，所以后面的时间要想满足条件，必须要在前面的条件满足的基础上进行计算，这就让算法的复杂度大大降低了
        {
            for(int i = 0 ; i < n ; i++)
            {
                for(int j = 0 ; j < n ; j++)
                {
                    if(f[i][j] >= f[i][k] + f[k][j])	//k是中转点
                        f[i][j] = f[i][k] + f[k][j];
                }
            }
            k++;
        }
        if(f[x][y] == inf || t[x] > tt || t[y] > tt)	//判断是否满足条件
            printf("-1\n");
        else
            printf("%d\n",f[x][y]);
    }
    return 0;
}