P1078 文化之旅

题目链接

搜索过的。
不过中间加上一个类似spfa里面的松弛优化。
数据好像比较水。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int M=107*107;
int n,m,s,t,k;
int b[109][109],c[109];
int vis[101];
int ans=1e9;
int dis[109][109],st[109],top;
int d[109];
void dfs(int u,int cost)
{
    if(cost>=ans) return;
    if(u==t)
    {
        if(ans>cost) ans=cost;
        return;
    } 
    for(int j=1;j<=n;j++)
    if(!vis[j]&&cost+dis[u][j]<ans&&d[j]>dis[u][j]+cost)
    {
        int flag=0;
        for(int i=1;i<=top;i++)
         if(vis[st[i]]&&b[c[j]][c[st[i]]])
         {
            flag=1;
            break;
         }
         if(flag) continue;
         d[j]=dis[u][j]+cost;
         vis[j]=1;
         st[++top]=j;
         dfs(j,cost+dis[u][j]); 
         vis[j]=0;
         top--;
    }
} 
int main()
{
    //freopen("a.txt","r",stdin);
    memset(dis,127/3,sizeof(dis));
    memset(d,127/3,sizeof(d));
    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<=k;i++)
      for(int j=1;j<=k;j++)
        scanf("%d",&b[i][j]);
    for(int i=1;i<=k;i++) b[i][i]=1;

    for(int i=1;i<=m;i++)
    {
        int u,v,d;
        scanf("%d%d%d",&u,&v,&d);
        dis[u][v]=min(d,dis[u][v]);
        dis[v][u]=min(d,dis[v][u]);
    }
    //memset(vis,0,sizeof(vis));
    vis[s]=1;st[++top]=s;d[s]=1;
    dfs(s,0);
    if(ans!=1e9) printf("%d",ans);
    else printf("-1\n");
    return 0;
}