[NOIP2012] 文化之旅 dfs

这道题就体现了聪明的搜索策略的重要性，如果我们正着搜，判断效率会明显下滑，所以我们就采用倒着搜索。(其实很玄学.....)

#include <cstdio>
#include <bitset>
#include <cstring>
#include <algorithm>
std::bitset<> now,keep[];
const int Inf=0x7f7f7f7f;
int n,m,s,t,k;
bool v[];
int dis[][],c[],ans=Inf;
int ch[][],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<> temp,int d){
    if(d>=ans)return;
    if(x==s){
        ans=Min(ans,d);
        return;
    }
    temp|=keep[c[x]],v[x]=;
    for(int i=;i<=n;i++)
        if(v[ch[x][i]]==&&dis[x][ch[x][i]]!=Inf&&temp[c[ch[x][i]]]==)
            dfs(ch[x][i],temp,d+dis[x][ch[x][i]]);
    v[x]=;
}
int main(){
    memset(dis,0x7f,sizeof(dis));
    scanf("%d%d%d%d%d",&n,&k,&m,&s,&t);
    for(int i=;i<=n;i++)scanf("%d",&c[i]);
    for(int i=;i<=n;i++){
        for(int j=;j<=n;j++)
            ch[i][j]=j;
        dis[i][i]=;
    }
    for(int i=,x;i<=k;i++){
        for(int j=;j<=k;j++)
            scanf("%d",&x),keep[i][j]=x;
        keep[i][i]=;
    }
    for(int i=,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=;i<=n;i++)
        ha=i,std::sort(ch[i]+,ch[i]+n+,comp);
    dfs(t,now,);
    printf("%d",ans==Inf?-:ans);
}