洛谷 P1119 灾后重建 （Floyd）

题目背景

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

题目描述

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

输入格式

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

第二行包含NN个非负整数t_0, t_1,…, t{N-1}，表示了每个村庄重建完成的时间，数据保证了t_0 ≤ t_1 ≤ … ≤ t_{N-1}。

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

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

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

输出格式

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

输入输出样例

输入 #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
-1
5
4

说明/提示

对于30%的数据，有N≤50；

对于30%的数据，有t_i= 0，其中有20%的数据有t_i = 0且N>50；

对于50%的数据，有Q≤100；

对于100%的数据，有N≤200，M≤N×(N−1)/2，Q≤50000，所有输入数据涉及整数均不超过100000。

理解Floyd算法

for (k = 1; k <= n; k++)
{
	for (i = 1; i <= n; i++)
    {
    	for (j = 1; j <= n; j++)
    	{
      		f[i][j] = min(f[i][j], f[i][k] + f[k][j]);
    	}
  	}
}

k在最外层，表示从i号顶点到j号顶点只经过前k号点的最短路程。

思路

按照时间顺序更新每一个可用的点(k)，对每一个可用的点进行Floyd算法内两层的循环(i , j)。

求出在当前时间的两点的最短路。

代码

#include <iostream>
#include <cstdio>
using namespace std;
#define MAXN 205
#define INF 30000050

int n, m, N;
int T[MAXN];
int dis[MAXN][MAXN];

void floyd(int k)
{
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < n; j++)
        {
            if (i != j && i != k && j != k)
            {
                if (dis[i][j] > dis[i][k] + dis[k][j])
                {
                    dis[i][j] = dis[i][k] + dis[k][j];
                }
            }
        }
    }
}

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)
                dis[i][j] = INF;
            else 
                dis[i][j] = 0;
        }
    }

    for (int i = 1; i <= m; i++)
    {
        static int x, y, z;
        cin >> x >> y >> z;
        dis[x][y] = dis[y][x] = z;
    }

    cin >> N;
    for (int i = 1, now = 0; i <= N; i++)
    {
        static int x, y, t;
        cin >> x >> y >> t;

        if (T[x] > t || T[y] > t)
        {
            cout << "-1" << endl;
            continue;
        }

        while (T[now] <= t && now < n)
        {
            floyd(now);
            now++;
        }

        if (dis[x][y] == INF)
            cout << "-1" << endl;
        else
            cout << dis[x][y] << endl;
    }
    return 0;
}