DS-图的最短路径-优先队列优化版

Description

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

Input

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

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

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

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

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

以此类推输入下一个示例

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

Output

对每组测试数据，输出：

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

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

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<iostream>
#include<queue>
#include<map>
using namespace std;
string s[1000],st;
int t,n,g[1000][1000],vis[1000],dis[1000],be[1000];
map<string,int>mp;
priority_queue<pair<int,int> >q;
void find(int index){
	if(be[index]>=0) find(be[index]);
	cout << s[index] << ' ';
}
void dj(int ss){
	dis[ss] = 0;
	q.push(make_pair(0,-ss));
	be[ss] = -1;
	while(!q.empty()){
		int index = -1*q.top().second;
		q.pop();
		if(vis[index]) continue;
		vis[index] = 1;
 		for(int i=0;i<n;i++){
			if(dis[index]+g[index][i]<dis[i]&& !vis[i]&&g[index][i]){
				be[i] = index;
				dis[i] = dis[index]+g[index][i];
				q.push(make_pair(-1*dis[i],-1*i));
			}
		}
	}
	for(int i=0;i<n;i++){
		if(i==ss) continue;
		if(dis[i]==1e6) cout << s[ss] << '-' << s[i] << "--1" << endl;
		else{
			cout << s[ss] << '-' << s[i] << "-" << dis[i] << "----[";
			find(i);
			cout << ']' << endl;
		}
	}
}
int main(){
	cin >> t;
	while(t--){
		cin >> n;
		for(int i=0;i<n;i++){
			cin >> s[i];
			mp[s[i]] = i;
			vis[i] = 0;
			dis[i] = 1e6;
		}
		for(int i=0;i<n;i++) for(int j=0;j<n;j++) cin >> g[i][j];
		cin >> st;
		dj(mp[st]);
	}
	return 0;
}

 

q.push(make_pair(-1*dis[i],-1*i));大的数变负数就会越小。

所以可以用大根堆去代替小根堆（说实话小根堆的写法不太好记住）。

be这个数组是用来打印的，用的是递归。

核心代码没几行，自己看就行。