文化之旅

QAQ
DFS 爆搜+剪枝

#include <cstdio>
#include <iostream>
#include <cstring>
#include <queue>
using namespace std;
bool vis[10001];
bool ok[10001][10001];
int dis[10001],w[1001];
int head[200000],net[200000],to[200000],cost[200000],cnt;
int n,k,s,t,m;
int ans=1e8;
queue <int> dl;
int wx[999],cntx;
void add(int x,int y,int z)
{
    cnt++;
    cost[cnt]=z;
    to[cnt]=y;
    net[cnt]=head[x];
    head[x]=cnt;
}
bool check(int q)
{
    int xx=w[q];
    //if(vis[xx]) return 0;
    for(int i=1;i<=cntx;i++)
     if(ok[xx][wx[i]]) return 0;
    return 1;
}
void dfs(int x,int Dis)
{
    //printf("%d\n",x);
    if(x==s)
    {
        ans=min(Dis,ans);
        return;
    }
    for(int i=head[x];i;i=net[i])
    {
        int p=to[i];
        if(dis[x]+cost[i]>=dis[p]||vis[w[p]]) continue;
        dis[p]=dis[x]+cost[i];
        if(check(p)) vis[w[p]]=1,wx[++cntx]=w[p],dfs(p,Dis+cost[i]),vis[w[p]]=0,cntx--;
    }
}
int main()
{
    memset(dis,127/3,sizeof(dis));
    scanf("%d%d%d%d%d",&n,&k,&m,&t,&s);
    for(int i=1;i<=n;i++)
     scanf("%d",&w[i]);
    for(int i=1;i<=k;i++)
     for(int j=1;j<=k;j++)
      scanf("%d",&ok[i][j]);
    for(int i=1;i<=m;i++)
     {
        int u,v,d;
        scanf("%d%d%d",&u,&v,&d);
        add(u,v,d),add(v,u,d);
     }
    memset(vis,0,sizeof(vis));
    dis[t]=0;
    wx[1]=w[t];cntx=1;
    vis[w[t]]=1;
    dfs(t,0);
    //printf("%d",dis[t]);
    if(ans==1e8) ans=-1;
    //vis[1]=1;
    printf("%d",ans);

    return 0;
}