P1119 災後重建 floyd

P1119 災後重建 floyd

好久沒寫floyd都快忘記這個有這個算法了

做法： 跑floyd不斷縮距離，floyd的核心是對$k=[1,n]$的點暴力枚舉作爲中轉點，優化距離$d[i][j]=min(d[i][j],d[i][k]+d[k][j])$

但這題多了一個時間限制，也就是說這個中轉點的修復時間必須小於等於t時刻，由於這個t是遞增的，所以d[][]是不斷覆蓋的，簡單很多

注意事項：一開始我用最小生成樹+堆優化跑少於q次的最短路，雖然已經優化到極致，但是發現q次詢問中這個最短路跑的還是很多。。結果超了0.2s，然後瞄了一眼題解發現要用floyd做，而且INF一開始設置爲了1e9，但是如果幾個1e9相加好像會產生負數？導致最後找了好久的錯誤，服了

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<queue>
#include<map>
#define ll long long
#define pb push_back
#define rep(x,a,b) for (int x=a;x<=b;x++)
#define repp(x,a,b) for (int x=a;x<b;x++)
#define W(x) printf("%d\n",x)
#define WW(x) printf("%lld\n",x)
#define pi 3.14159265358979323846
#define mem(a,x) memset(a,x,sizeof a)
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
using namespace std;
const int maxn=2e2+7;
const int INF=1e6;
const ll INFF=1e18;
int d[maxn][maxn];
bool vis[maxn];
int u[100007],v[100007],t[100007],tim[maxn],n,m,x,y,z,q;

int main()
{
    mem(d,INF);mem(vis,false);
    scanf("%d%d",&n,&m);
    rep(i,1,n)scanf("%d",&tim[i]),d[i][i]=0;
    rep(i,1,m)
    {
        scanf("%d%d%d",&x,&y,&z);
        x++;y++;
        d[x][y]=d[y][x]=z;
    }
    scanf("%d",&q);
    rep(i,1,q)
    {
        scanf("%d%d%d",&u[i],&v[i],&t[i]);
        u[i]++;v[i]++;
    }
    rep(l,1,q)
    {
        rep(k,1,n)
        {
            if (!vis[k]&&tim[k]<=t[l])
            {
                vis[k]=true;
                rep(i,1,n)
                    rep(j,1,n)
                        if (i!=j&&j!=k&&i!=k&&d[i][j]>d[i][k]+d[k][j]&&d[i][k]<INF&&d[k][j]<INF)
                            d[i][j]=d[i][k]+d[k][j];
            }
        }
        if (tim[u[l]]<=t[l]&&tim[v[l]]<=t[l]&&d[u[l]][v[l]]<INF)W(d[u[l]][v[l]]);
        else W(-1);
    }
    return 0;
}