[NOIP2012] 文化之旅 dfs

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

#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);
}
原文地址:https://www.cnblogs.com/TSHugh/p/7537937.html