代码。
#include<cstdio>
#include<vector>
using namespace std;
const int maxn=104;
bool vis[maxn];
int n,k,m,s,t,u,v,d,res=0x7fffffff,c[maxn],a[maxn][maxn];
struct edge{ int to,w;};
vector < edge > g[maxn];
inline void read(int &x){
char c=getchar();bool f=0;x=0;
for(;c<'0'||c>'9';c=getchar())
if(c=='-')
f=1;
for(;c>='0'&&c<='9';c=getchar())
x=x*10+(c^48);
x=f?-x:x;
}
void dfs(int node,int len){
if(len>=res)
return ;
if(node==t){ //到了就更新
res=len;
return ;
}
for(int i=g[node].size()-1;i>=0;i--){
edge e=g[node][i];int to=e.to;
if(vis[c[to]])
continue;
vis[c[to]]=1;
dfs(to,len+e.w);
vis[c[to]]=0;
}
}
signed main(){
read(n),read(k),read(m),read(s),read(t);
for(int i=1;i<=n;i++)
read(c[i]);
for(int i=1;i<=k;i++)
for(int j=1;j<=k;j++)
read(a[i][j]); //文化i排斥文化j
while(m--){
read(u),read(v),read(d);
if(!a[v][u]) //如果通往的国家不排斥现在的文化
g[u].push_back((edge){v,d});
if(!a[u][v]) //同理
g[v].push_back((edge){u,d});
}
vis[c[s]]=1;
dfs(s,0);
printf("%d",res==0x7fffffff?-1:res);
return 0;
}