dijkstra40分求助
查看原帖
dijkstra40分求助
863660
zoee123楼主2024/3/7 23:45
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int maxm=10005;
const int maxn=105;

struct node{
	int dis,u;
	bool operator>(const node& x) const {
		return dis>x.dis;
	}
};

int e[maxn][maxn];
int dis[maxn][maxn],vis[maxn],path[maxm];
priority_queue<node,vector<node>,greater<node>> q;

void dijkstra(int n,int s){
	memset(vis,0,sizeof(vis));
	memset(dis[s],63,sizeof(dis[s]));
	dis[s][s]=0;
	q.push({0,s});
	while(!q.empty()){
		int u=q.top().u;
		q.pop();
		if(vis[u]) continue;
		vis[u]=1;
		for(int i=1;i<=n;i++){
			int w=e[u][i];
			if(dis[s][i]>dis[s][u]+w){
				dis[s][i]=dis[s][u]+w;
				q.push({dis[s][i],i});
			}
		}
	}
}

int main(){
	int n,m,s,u,v,w,ans=0;
	scanf("%d %d",&n,&m);
	for(int i=1;i<=m;i++){
		scanf("%d",&path[i]);
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			scanf("%d",&e[i][j]);
		}
	}
	for(int i=1;i<=n;i++){
		dijkstra(n,path[i]);
	}
	for(int i=1;i<m;i++){
		ans+=dis[path[i]][path[i+1]];
		//printf("%d ",ans);
	}
	printf("%d",ans);
}
2024/3/7 23:45
加载中...