PTA L2 - 紧急救援

紧急救援

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

输入格式:
输入第一行给出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

AC代码

#include "bits/stdc++.h"
#define inf 0x3f3f3f3f
#define P pair<int, int>
using namespace std;
const int N = 510;
int city[N];
int dis[N];
int pre[N];
int road[N];
int peo[N];
vector<P> edge[N];
int main(){
	int n, m, s, d;
	cin>>n>>m>>s>>d; // city number is 0 -- n - 1
	for(int i = 0; i < n; i++){
		cin>>city[i];
		peo[i] = city[i];
	}
	for(int i = 0; i < m; i++){
		int x, y, l;
		cin>>x>>y>>l;
		edge[x].push_back(P(y, l));
		edge[y].push_back(P(x, l));
	}
	
	// 初始化数据
	memset(pre, -1, sizeof pre);
	memset(dis, inf, sizeof dis);
	
	priority_queue<P, vector<P>, greater<P> > q; // 优先队列实现Dijkstra算法, greater<P> 即
	                                             // 距离短排前面 
	q.push(P(0, s)); // P(distant, vertex) 
	dis[s] = 0;
	road[s] = 1;
	peo[s] = city[s];
	
	while(q.size()){
		P t = q.top(); q.pop();
		int dist = t.first;     // 距离 
		int cur = t.second;     // 起点 
		if(dis[cur] < dist) continue; // 如果当前节点已经得出结果那么该点没有继续的意义了，跳过
		for(P t1 : edge[cur]){
			int next = t1.first;
			int len = t1.second;
			if(dis[next] > dis[cur] + len){
				dis[next] = dis[cur] + len;
				road[next] = road[cur];
				pre[next] = cur;
				peo[next] = peo[cur] + city[next];
				q.push({dis[next], next});
			}
			else if(dis[next] == dis[cur] + len){ // 相同距离的最短路径 
				if(peo[next] < peo[cur] + city[next]){
					pre[next] = cur;
				}
				peo[next] = max(peo[next], peo[cur] + city[next]);
				road[next] += road[cur];
			}
		} 
	}
	
	
	cout<<road[d]<<" "<<peo[d]<<endl;
	
	vector<int> ans;
	while(true){
		ans.push_back(d);
		d = pre[d];
		if(d == -1)break;
	}
	reverse(ans.begin(), ans.end());
	for(int i = 0; i < ans.size(); i++){
		if(i == 0)cout<<ans[i];
		else cout<<" "<<ans[i];
	}
	
	return 0;
}