【vijos】【最短路變式】小D的旅行

描述

旅行是一件頗有趣的事情，但是在旅行前規劃好路線也很重要。
現在小D計劃要去U國旅行。
U國有N個城市，M條道路，每條道路都連接着兩個城市，並且經過這條道路需要一定的費用wi。
現在小D想要從u城市到v城市，但是他的汽車需要在途中加一次油(途中包括u和v兩個城市)。在每個城市加油都有不同的費用vi。
小D想知道從u城市到v城市最少需要多少費用（經過道路的費用+加油的費用）。
城市從1-n進行編號。

格式

輸入格式

第一行兩個正整數n，m，表示n個城市，m條無向道路
接下來n行，第i行一個整數vi，表示第i個城市的加油費用
接下來m行，第i行三個整數ai, bi, wi，表示第i條道路連接ai和bi兩個城市，經過要花費wi的費用
接下來一個正整數q，表示小D有q個詢問
接下來q行，第i行兩個整數ui, vi， 表示小D想知道從ui到vi需要的最少費用(ui和vi可能相等)

輸出格式

對於每個詢問，輸出一行整數，表示最小的費用，如果ui不能到達vi，則輸出-1

代碼

做兩遍floyd即可。
注意floyd的循環順序是k,i,j。

#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 310
#define INF 1000000000-1
using namespace std;

int n,m,q,a,b,w,u,l,v[maxn];
int dist[maxn][maxn],map[maxn][maxn];

int main(){
    //freopen("in.txt","r",stdin);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&v[i]);
    for(int i=1;i<=n;i++)
    {
        map[i][i]=v[i]; dist[i][i]=0;
        for(int j=i+1;j<=n;j++) map[i][j]=map[j][i]=dist[i][j]=dist[j][i]=INF;
    }
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&a,&b,&w);
        if(a==b) continue;
        dist[b][a]=dist[a][b]=min(dist[a][b],w);
    }
    for(int k=1;k<=n;k++)
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                dist[i][j]=min(dist[i][j],dist[i][k]+dist[k][j]);
    for(int k=1;k<=n;k++)
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                {
                    map[i][j]=min(map[i][j],dist[i][k]+dist[k][j]+v[k]);
                }
    scanf("%d",&q);
    while(q--){
        int ans=INF;
        scanf("%d%d",&u,&l);
        ans=map[u][l];
        if(ans<INF) printf("%d\n",ans);
        else printf("-1\n");
    }
    return 0;
}