提交记录

#include <iostream>
using namespace std;
const int N=105,inf=1e9;
int n,k,m,s,t,c[N],a[N][N],e[N][N],u,v,d;
int main(){
    scanf("%d%d%d%d%d",&n,&k,&m,&s,&t);
    for(int i=1;i<=n;i++){
        scanf("%d",&c[i]);
        for(int j=1;j<=n;j++) e[i][j]=inf;
        e[i][i]=0;
    }
    for(int i=1;i<=k;i++){
        for(int j=1;j<=k;j++){
            scanf("%d",&a[i][j]);
        }
    }
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&u,&v,&d);
        if(!a[c[u]][c[v]]) e[v][u]=min(e[v][u],d);
        if(!a[c[v]][c[u]]) e[u][v]=min(e[u][v],d);
    }
    for(int j=1;j<=n;j++){
        for(int i=1;i<=n;i++){
            for(int k=1;k<=n;k++){
                if(e[i][k]>e[i][j]+e[j][k]){
                    e[i][k]=e[i][j]+e[j][k];
                }
            }
        }
    }
    if(e[s][t]==inf) printf("-1");
    else printf("%d",e[s][t]);
    return 0;
}