Luogu P1078 文化之旅

Luogu P1078 文化之旅

一打开题目，你就会看见第一行有这样几个字——“本题是错题”。但这并不影响此题的价值以及CCF的黑心。
首先，因为数据小，所以用邻接矩阵存图即可。
其次，如果起点和终点的文化相同，那显然无解。
最后，跑一遍Floyd，如果起点到终点的距离仍然为$INF$，那就说明起点到终点没有通路；否则输出距离即可。
（插一句，上次某题因为$INF$设成$0x7fffffff$而玄学出错，所以现在$INF$也只敢设成$0x3f3f3f3f$了）

#include<bits/stdc++.h>
#define N 110
#define INF 0x3f3f3f3f

using namespace std;

int n,k,m,s,t;
int edge[N][N],repel[N][N],c[N];

void Init() {
	for(int i=1;i<=n;i++) {
		for(int j=1;j<=n;j++) {
			edge[i][j]=INF;
		}
		edge[i][i]=0;
	}
	return;
}

void addEdge(int u,int v,int w) {
	if(!repel[c[v]][c[u]]&&c[v]!=c[u]) {
		edge[u][v]=min(w,edge[u][v]);
	}
	if(!repel[c[u]][c[v]]&&c[u]!=c[v]) {
		edge[v][u]=min(w,edge[v][u]);
	}
	return;
}

void Read() {
	scanf("%d%d%d%d%d",&n,&k,&m,&s,&t);
	Init();
	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",&repel[i][j]);
		}
	}
	for(int i=1;i<=m;i++) {
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		addEdge(u,v,w);
	}
	return;
}

void Floyd() {
	for(int k=1;k<=n;k++) {
		for(int i=1;i<=n;i++) {
			for(int j=1;j<=n;j++) {
				edge[i][j]=min(edge[i][j],edge[i][k]+edge[k][j]);
			}
		}
	}
	return;
}

void Solve() {
	if(c[s]==c[t]) {
		printf("-1");
		return;
	}
	Floyd();
	if(edge[s][t]==INF) {
		printf("-1");
	}
	else {
		printf("%d",edge[s][t]);
	}
	return;
}

int main()
{
	Read();
	Solve();
	return 0;
}