深搜,剪枝。个人认为此题与棋盘其实非常相似
用一个数组记录到达某个国家的最小距离(初值无穷大),就可以把大于等于该距离的剪枝掉(注意等于也必须减去,否则极有可能玄学错误)
#include<bits/stdc++.h>
using namespace std;
int n,k,m,s,t,ans=0x3f3f3f3f;
int cou[107],cul[107][107],dis[107][107];
// 国家的文化 文化间排斥的表 国家之间的距离
bool vis[107];int jud[107];
// 已经学过的文化 到达某个国家的最小距离
void dfs(int now,int sum){
// 现在在的国家 已走的距离
if(now==t){ans=min(ans,sum);return;}//到达目的地
if(sum>=jud[now])return;else jud[now]=sum;
//剪枝,记得要更新最小值
for(int i=1;i<=n;++i){
if(!vis[cou[i]]&&dis[now][i]<0x3f3f3f3f&&sum<jud[i]){//没去过,有道路,不能够剪枝
bool flag=false;
for(int j=1;j<=k;++j){
if(cul[cou[i]][j]&&vis[j]){flag=true;break;}//看是否有排斥现象
}
if(flag)continue;//有排斥
vis[cou[i]]=true;//学习文化
dfs(i,sum+dis[now][i]);//下一站
vis[cou[i]]=false;//记得回溯
}
}
return;
}
int main(){
memset(dis,0x3f,sizeof(dis));
memset(cul,0,sizeof(cul));//此处建议不要设成-1,因为个人的写法不同(if),防止有问题(数组越界了之类,虽然基本不可能,但尽量减少错误机会)
memset(jud,0x3f,sizeof(jud));
cin>>n>>k>>m>>s>>t;
for(int i=1;i<=n;++i)cin>>cou[i];
for(int i=1;i<=k;++i)for(int j=1;j<=k;++j)cin>>cul[i][j];
for(int i=1;i<=m;++i){
int u,v,d;//在循环里定义
cin>>u>>v>>d;
dis[u][v]=min(dis[u][v],d);//记得取小,因为可能有多条道路
}
vis[cou[s]]=true;//出发点的文化别忘记学
dfs(s,0);
if(ans!=0x3f3f3f3f)cout<<ans<<endl;
else cout<<"-1"<<endl;//ans没变,说明没有到达
return 0;
}