DS图—图的最短路径（不含代码框架）

题目描述

给出一个图的邻接矩阵，输入顶点v，用迪杰斯特拉算法求顶点v到其它顶点的最短路径。

输入

第一行输入t，表示有t个测试实例

第二行输入顶点数n和n个顶点信息

第三行起，每行输入邻接矩阵的一行，以此类推输入n行

第i个结点与其它结点如果相连则为距离，无连接则为0，数据之间用空格

隔开。第四行输入v0，表示求v0到其他顶点的最短路径距离

以此类推输入下一个示例

输出

对每组测试数据，输出：

每行输出v0到某个顶点的最短距离和最短路径

每行格式：v0编号-其他顶点编号-最短路径值----[最短路径]。没有路径输出：v0编号-其他顶点编号--1。具体请参考示范数据

 样例输入

2
5 0 1 2 3 4
0 5 0 7 15
0 0 5 0 0
0 0 0 0 1
0 0 2 0 0
0 0 0 0 0
0
6 V0 V1 V2 V3 V4 V5
0 0 10 0 30 100
0 0 5 0 0 0
0 0 0 50 0 0
0 0 0 0 0 10
0 0 0 20 0 60
0 0 0 0 0 0
V0

样例输出

0-1-5----[0 1 ]
0-2-9----[0 3 2 ]
0-3-7----[0 3 ]
0-4-10----[0 3 2 4 ]
V0-V1--1
V0-V2-10----[V0 V2 ]
V0-V3-50----[V0 V4 V3 ]
V0-V4-30----[V0 V4 ]
V0-V5-60----[V0 V4 V3 V5 ]

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=100,INF=0x3f3f3f3f;
int t;
map<string,int> mp;
map<int,string> ins;

class GRAPH{
public:
	int n,w[maxn][maxn],dist[maxn],st;
	bool vis[maxn];
	string path[maxn];
	GRAPH(){
		cin>>n;
		for(int i=0;i<n;i++){
			string x;
			cin>>x;
			mp[x]=i;
			ins[i]=x;
		}
		for(int i=0;i<n;i++){
			vis[i]=false;
			dist[i]=INF;
			for(int j=0;j<n;j++){
				cin>>w[i][j];
			}
		}
	}
	void getpath(){
		string start;
		cin>>start;
		st=mp[start];
		dist[st]=0;
		for(int i=0;i<n;i++){
			path[i]+=ins[st]+" ";
			if(i!=st && w[st][i]){
				path[i]+=ins[i]+" ";
				dist[i]=w[st][i];
			}
		}
		vis[st]=true;
		for(int k=0;k<n-1;k++){
			int res,dis=INF;
			for(int i=0;i<n;i++){
				if(!vis[i] && dis>dist[i]){
					res=i;
					dis=dist[i];
				}
			}
			
			for(int i=0;i<n;i++){
				if(!vis[i] && dist[i]>dist[res]+w[res][i] && w[res][i]){
					dist[i]=dist[res]+w[res][i];
					path[i]=path[res];
					path[i]+=ins[i]+" ";
				}
			}
			vis[res]=true;
		}
	}
	
	void display(){
		for(int i=0;i<n;i++){
			if(i==st) continue;
			if(dist[i]==INF){
				dist[i]=-1;
				cout<<ins[st]<<"-"<<ins[i]<<"-"<<dist[i]<<endl;
			}
			else cout<<ins[st]<<"-"<<ins[i]<<"-"<<dist[i]<<"----["<<path[i]<<"]\n";
		}
	}
};

int main(){
	cin>>t;
	while(t--){
		GRAPH g;
		g.getpath();
		g.display();
	}
	return 0;
}