洛谷 P1078文化之旅题解--zhengjun

题面传送门

思路

一看，数据这么小，打个搜索剪枝一下就可以了。

然后，我打完了之后，$T$了一个点，就使劲想剪枝，就是已经走到的点就不用再走一次了。

代码

#include<bits/stdc++.h>
#define maxn 101
#define maxm 10001
using namespace std;
int n,k,m,s,t;
int c[maxn];
int a[maxn][maxn];
int head[maxn],to[maxm],v[maxm],nex[maxm],tot;
void add(int x,int y,int z){
	to[tot]=y;v[tot]=z;
	nex[tot]=head[x];
	head[x]=tot++;
}
int f[maxn],vis[maxn];
int ans=0x3fffffff;
void dfs(int x,int sum){
	if(sum>=ans)return;
	if(x==t)ans=sum;
	if(vis[x])return;
	vis[x]=1;
	f[c[x]]++;
	for(int pos=head[x];pos!=-1;pos=nex[pos]){
		bool flag=1;
		for(int i=1;i<=k;i++){
			if(f[i]>0&&a[i][c[to[pos]]]==1){
				flag=0;
				break;
			}
		}
		if(flag)
			dfs(to[pos],sum+v[pos]);
	}
	f[c[x]]--;
}
int main(){
	memset(head,-1,sizeof(head));
	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",&a[i][j]);
		a[i][i]=1;
	}
	int x,y,z;
	while(m--){
		scanf("%d%d%d",&x,&y,&z);
		add(x,y,z);
		add(y,x,z);
	}
	dfs(s,0);
	if(ans<0x3fffffff)printf("%d",ans);
	else printf("-1");
	return 0;
}

谢谢--zhengjun