dij,80tps,求助!
查看原帖
dij,80tps,求助!
907430
corner_xiejunqi楼主2024/10/6 09:28

#include<bits/stdc++.h>
using namespace std;
int n,k,m,s,t;
int culture[101];
int dislike[101][101];
int g[101][101];
int dis_country[101][101];
int dis[101];
int book[101];
int main(){
	//1、声明变量,输入
	cin>>n>>k>>m>>s>>t;
	for(int i=1;i<=n;i++){
		cin>>culture[i];
	}
	for(int i=1;i<=k;i++){
		for(int j=1;j<=k;j++){
			cin>>dislike[i][j];
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			dis_country[i][j]=dislike[culture[i]][culture[j]];
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			if(i==j) g[i][j]=0;
			else g[i][j]=0x3f3f3f3f;
		}
	}
	for(int i=1;i<=m;i++){
		int u,v,d;
		cin>>u>>v>>d;
		g[u][v]=g[v][u]=d;
	}
	//2、计算过程
	for(int i=1;i<=n;i++){
		if(dis_country[i][s]) dis[i]=0x3f3f3f3f;
		else dis[i]=g[s][i];
	}
	book[s]=1;
	for(int i=1;i<=n-1;i++){
		int minn=0x3f3f3f3f,u;
		for(int j=1;j<=n;j++){
			if(!book[j] && dis[j]<minn && dis_country[j][s]==0){
				minn=dis[j];
				u=j;
			}
		}
		book[u]=1;
		for(int j=1;j<=n;j++){
			if(!book[j] && dis[u]+g[u][j]<dis[j] && dis_country[j][u]==0){
				dis[j]=dis[u]+g[u][j];
			}
		}
	}
	//3、输出
	if(dis[t]==0x3f3f3f3f) cout<<-1;
	else cout<<dis[t];
	return 0;
}


2024/10/6 09:28
加载中...