这道题就体现了聪明的搜索策略的重要性,如果我们正着搜,判断效率会明显下滑,所以我们就采用倒着搜索。(其实很玄学.....)
#include <cstdio> #include <bitset> #include <cstring> #include <algorithm> std::bitset<110> now,keep[110]; const int Inf=0x7f7f7f7f; int n,m,s,t,k; bool v[110]; int dis[110][110],c[110],ans=Inf; int ch[110][110],ha; inline bool comp(int x,int y){ return dis[ha][x]<dis[ha][y]; } inline int Min(int x,int y){ return x<y?x:y; } void dfs(int x,std::bitset<110> temp,int d){ if(d>=ans)return; if(x==s){ ans=Min(ans,d); return; } temp|=keep[c[x]],v[x]=1; for(int i=1;i<=n;i++) if(v[ch[x][i]]==0&&dis[x][ch[x][i]]!=Inf&&temp[c[ch[x][i]]]==0) dfs(ch[x][i],temp,d+dis[x][ch[x][i]]); v[x]=0; } int main(){ memset(dis,0x7f,sizeof(dis)); 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<=n;i++){ for(int j=1;j<=n;j++) ch[i][j]=j; dis[i][i]=0; } for(int i=1,x;i<=k;i++){ for(int j=1;j<=k;j++) scanf("%d",&x),keep[i][j]=x; keep[i][i]=1; } for(int i=1,x,y,z;i<=m;i++) scanf("%d%d%d",&x,&y,&z),dis[x][y]=dis[y][x]=Min(dis[x][y],z); for(int i=1;i<=n;i++) ha=i,std::sort(ch[i]+1,ch[i]+n+1,comp); dfs(t,now,0); printf("%d",ans==Inf?-1:ans); }