【PAT天梯赛】L2-001 紧急救援 (25分) C++实现 PAT团队天梯赛 Dijkstra求单元最短路径

题目

作为一个城市的应急救援队伍的负责人，你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候，你的任务是带领你的救援队尽快赶往事发地，同时，一路上召集尽可能多的救援队。

输入格式:
输入第一行给出4个正整数N、M、S、D，其中N（2≤N≤500）是城市的个数，顺便假设城市的编号为0 ~ (N−1)；M是快速道路的条数；S是出发地的城市编号；D是目的地的城市编号。

第二行给出N个正整数，其中第i个数是第i个城市的救援队的数目，数字间以空格分隔。随后的M行中，每行给出一条快速道路的信息，分别是：城市1、城市2、快速道路的长度，中间用空格分开，数字均为整数且不超过500。输入保证救援可行且最优解唯一。

输出格式:
第一行输出最短路径的条数和能够召集的最多的救援队数量。第二行输出从S到D的路径中经过的城市编号。数字间以空格分隔，输出结尾不能有多余空格。

输入样例:
4 5 0 3
20 30 40 10
0 1 1
1 3 2
0 3 3
0 2 2
2 3 2
输出样例:
2 60
0 1 3

思路

使用Dijkstra算法计算单源的最短路径，一开始我看这个题任意两点以为使用Floyd算法做结果有测试点超时了（可能是之前刷acm题然后会给很多组数据，然后就是多源了，pat的题目好像都是单组测试的吧）。。后来才发现原来用Dijkstra算法也可以，只要把起点设置好就行了- -。。

代码

//看了其他小伙伴的代码改的，优先队列优化版本的Dijkstra算法
#include<bits/stdc++.h>
#include<functional>
using namespace std;
typedef pair<int, int> P;
vector<vector<P > > v;
int n, m, s, d;
int cnt[505], dis[505], last[505], curSum[505], path[505];
bool vis[505];
void dijkstra() {
	for(int i = 0; i < 505; i++) {
		dis[i] = INT_MAX >> 1;
		vis[i] = false;
		last[i] = -1;
		curSum[i] = cnt[i];
		path[i] = 1;
	}
	dis[s] = 0;
	priority_queue<P, vector<P>, greater<P> > q;
	q.push(make_pair(0, s));
	while(q.size()) {
		P e = q.top();
		q.pop();
		int z = e.first, y = e.second;
		if(vis[y] == true) continue;
		vis[y] = true;
		for(int i = 0; i < v[y].size(); i++) {
			int x = v[y][i].second, z = v[y][i].first;
			if(dis[y] + z < dis[x]) {
				curSum[x] = curSum[y] + cnt[x];
				dis[x] = dis[y] + z;
				last[x] = y;
				q.push(make_pair(dis[x], x));
				path[x] = path[y];
			} else if(dis[x] == dis[y] + z) {
				path[x] = path[x] + path[y];
				if(curSum[x] < curSum[y] + cnt[x]) {
					curSum[x] = curSum[y] + cnt[x];
					last[x] = y;
					q.push(make_pair(dis[x], x));
				}
			}
		}
	}
}
int main() {
	cin >> n >> m >> s >> d;
	for(int i = 0; i < n; i++) cin >> cnt[i];
	v.resize(n);
	for(int i = 0; i < m; i++) {
		int x, y, z;
		cin >> x >> y >> z;
		v[x].push_back(make_pair(z, y));
		v[y].push_back(make_pair(z, x));
	}
	dijkstra();
	cout << path[d] << " " << curSum[d] << endl;
	vector<int > ans;
	int g = d;
	while(g != -1) {
		ans.push_back(g);
		g = last[g];
	}
	for(int i = ans.size() - 1; i >= 0; i--) {
		if(i < ans.size() - 1) cout << " ";
		cout << ans[i];
	}
	return 0;
}