P1078 文化之旅

吐槽

为什么一些明显不对的算法也过了！！！！！！！！！！！

思路

明显是最短路。加一个文化判断即可。

#include<cstdio>
#include<cstring>
int min(int x,int y){return x<y?x:y;}

int c[105],n;//c数组存文化种类
int a[105][105];//a数组存是否排斥
int f[105][105];//f数组存路径长度
bool used[105][105][105];//i,j,k表示i向j的边经过了文化k
int gg;
void floyd()//求最短路并判断是否排斥或重复
{
    for(int k=1;k<=n;k++)
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                if(!a[c[k]][c[i]]&&!a[c[j]][c[k]]&&!used[i][k][c[j]]&&!used[k][j][c[i]]&&f[i][k]+f[k][j]<f[i][j])
                {
                    
                    used[i][j][c[k]]=true;
                    f[i][j]=f[i][k]+f[k][j];
                }
}
int main()
{
    memset(f,0x3f,sizeof(f));
    int k,m,s,t,u,v,w;
    scanf("%d%d%d%d%d",&n,&k,&m,&s,&t);
    gg=k;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&c[i]);
        f[i][i]=0;//最短路初始化
    }
    for(int i=1;i<=k;i++)
        for(int j=1;j<=k;j++)
            scanf("%d",&a[i][j]);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&u,&v,&w);
        if(!a[c[v]][c[u]]&&c[u]!=c[v])//如果两个点排斥或文化相同则不能连边
            f[u][v]=min(w,f[u][v]);//两个点之间不止一条路
        if(!a[c[u]][c[v]]&&c[u]!=c[v])
            f[v][u]=min(w,f[v][u]);
    }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        {
            used[i][j][c[i]]=true;//初始化条件
            used[i][j][c[j]]=true;
        }
    floyd();
    if(f[s][t]==0x3f3f3f3f)
        printf("-1\n");//不能到达
    else
        printf("%d\n",f[s][t]);
    return 0;
}