P1119 災後重建(Floyd)

題目來源

P1119 災後重建

---

解題思路

• 本題可用dijkstra做，對於每個時間限制都輸入dijkstra進行判斷，但時間複雜度高，需要剪枝、O2優化等操作才能勉強通過，不是本題正解
• 本題最大結點數爲200符合floyd算法的計算範圍
• 輸入每個村莊重建完成的時間，保證是不下降的，且每次詢問的時間也是不下降的
• 則可以對現在的k輸入進floyd進行對每兩個結點間距離的更新(即以這個點作爲中間點，對兩點之間的最短距離進行更新，符合floyd算法內涵)
• 注意路是雙向的，賦值時需要對兩個賦值
• 自己到自己結點的距離要初始化爲0

---

代碼實現

//floyd正解
#include<bits/stdc++.h>
using namespace std;
const int MAXN = 1000;
const int INF = 0x3f3f3f3f;

int t[MAXN];
int N, M, Q;
int d[MAXN][MAXN];

void floyd(int k)
{
    for(int i = 0; i < N; i++)
    {
        for(int j = 0; j < N; j++)
        {
            if(d[i][k] != INF && d[k][j] != INF && d[i][k]+d[k][j] < d[i][j])
            {
                d[j][i] = d[i][j] = d[i][k]+d[k][j];  //雙向邊更新
            }
        }
    }
}
int main()
{
    fill(d[0], d[0]+MAXN*MAXN, INF);
//    freopen("in.txt","r",stdin);
    scanf("%d%d",&N,&M);
    for(int i = 0; i < N; i++)
    {
        scanf("%d",&t[i]);
        d[i][i] = 0;      //初始化
    }
    for(int i = 0; i < M; i++)
    {
        int u, v, w;
        scanf("%d%d%d",&u,&v,&w);
        d[u][v] = d[v][u] = w;
    }
    scanf("%d",&Q);
    int now = 0;         //當前作爲更新的點
    for(int i = 0; i < Q; i++)
    {
        int x, y, t0;
        scanf("%d%d%d",&x,&y,&t0);
        for( ; t[now] <= t0 && now < N; now++) //更新
        {
            floyd(now);
        }
        if(t[x]>t0 || t[y]>t0 || d[x][y] == INF)
            printf("-1\n");
        else
            printf("%d\n",d[x][y]);
    }
    return 0;
}