蒟蒻60分,哪位大佬帮帮我???
查看原帖
蒟蒻60分,哪位大佬帮帮我???
658786
STUDENT00楼主2022/9/12 17:17

代码很好理解!!!

#include<bits/stdc++.h>
using namespace std;
int n,k,m,s,t,u,v,d,c[110],a[110][110],num[110],w[110][110],ans=1e9;
bool f[110][110],vis[110],ff[110];
void dfs(int now,int sum){
	if(now==t){
		ans=min(ans,sum);
		return;
	}
	for(int i=1;i<=num[now];i++){
		if(ff[a[now][i]]||vis[c[a[now][i]]]) continue;
		bool flag=1;
		for(int j=1;j<=k;j++){
			if(vis[j]&&f[c[a[now][i]]][j]){
				flag=0;
				break;
			}
		}
		if(flag){
			vis[c[a[now][i]]]=ff[a[now][i]]=1;
			dfs(a[now][i],sum+w[now][a[now][i]]);
			vis[c[a[now][i]]]=ff[a[now][i]]=0;
		}
	}
}
int main(){
	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",&f[i][j]);
	}
	while(m--){
		scanf("%d%d%d",&u,&v,&d);
		a[u][++num[u]]=v;
		a[v][++num[v]]=u;
		w[u][v]=w[v][u]=d;
	}
	vis[c[s]]=ff[s]=1;
	dfs(s,0);
	if(ans==1e9) printf("-1");
	else printf("%d",ans);
	return 0;
}
2022/9/12 17:17
加载中...