思路
一看,数据这么小,打个搜索剪枝一下就可以了。
然后,我打完了之后, T T T了一个点,就使劲想剪枝,就是已经走到的点就不用再走一次了。
代码
#include<bits/stdc++.h>
#define maxn 101
#define maxm 10001
using namespace std;
int n,k,m,s,t;
int c[maxn];
int a[maxn][maxn];
int head[maxn],to[maxm],v[maxm],nex[maxm],tot;
void add(int x,int y,int z){
to[tot]=y;v[tot]=z;
nex[tot]=head[x];
head[x]=tot++;
}
int f[maxn],vis[maxn];
int ans=0x3fffffff;
void dfs(int x,int sum){
if(sum>=ans)return;
if(x==t)ans=sum;
if(vis[x])return;
vis[x]=1;
f[c[x]]++;
for(int pos=head[x];pos!=-1;pos=nex[pos]){
bool flag=1;
for(int i=1;i<=k;i++){
if(f[i]>0&&a[i][c[to[pos]]]==1){
flag=0;
break;
}
}
if(flag)
dfs(to[pos],sum+v[pos]);
}
f[c[x]]--;
}
int main(){
memset(head,-1,sizeof(head));
scanf("%d%d%d%d%d",&n,&k,&m,&s,&t);
for(int i=1;i<=n;i++)scanf("%d",&c[i]);
for(int i=1;i<=k;i++){
for(int j=1;j<=k;j++)
scanf("%d",&a[i][j]);
a[i][i]=1;
}
int x,y,z;
while(m--){
scanf("%d%d%d",&x,&y,&z);
add(x,y,z);
add(y,x,z);
}
dfs(s,0);
if(ans<0x3fffffff)printf("%d",ans);
else printf("-1");
return 0;
}