团体程序设计天梯赛-练习集-L2-001 紧急救援（dij最短路）

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

其实是比较裸的最短路问题，但是2018年天梯赛前练习写到这题时，发现不会，照着网上博客写了一篇，然后想着写篇题解，结果写题解的时候发现自己又不会了。。。于是一直咕咕咕。。。（现在依然很菜的我想不通自己当时怎么会那么菜，还自我感觉不错。。。）

最近又开始准备天梯赛，写到了这题，写完之后突然想起了草稿箱的这篇博客，于是补完。

最短路我稍微熟悉一点的就是迪杰斯特拉算法，本题比原始的最短路问题麻烦一点的地方在于求最短路的条数和求出其中点权和最大的一条最短路。

首先，我们知道迪杰斯特拉算法中最重要的部分在于松弛操作，求任意一条最短路的话只需要在dis[u]+w<dis[v]更新即可。

求条数则需要用到类似dp的思想，当dis[u]+w<dis[v]时，rnum[v]=rnum[u]；当dis[u]+w==dis[v]时，说明从当前的u到v同样也是s到v的最短路，则有rnum[v]+=rnum[u]。

求最短路中点权和最大的路径，则当dis[u]+w<dis[v]时，无条件更新pnum[v]=pnum[u]+num[v]，pre[v]=u；当dis[u]+w==dis[v]时，若pnum[u]+num[v]>pnum[v]，则更新pnum[v]=pnum[u]+num[v]，pre[v]=u。

其余做法与裸的迪杰斯特拉算法一样。

最后通过pre[d]找到完整的路径，输出答案即可。

代码如下：

#include<bits/stdc++.h>
using namespace std;
#define p_b push_back
#define For(i,a,b) for(int i=a;i<=b;i++)
#define ll long long
const int N=5e2+5;
const int M=15;
#define INF 0x3f3f3f3f
int n,m,s,d;
struct Edge{
	int v,w;
	Edge(int _v=0,int _w=0){
		v=_v,w=_w;
	}
	friend bool operator <(Edge a,Edge b){
		return a.w>b.w;
	}
}tmp;
priority_queue<Edge> pq;
int dis[N],pnum[N],rnum[N],num[N],pre[N];
bool vis[N];
int u,v,w;
vector<Edge> e[N];
void dij(){
	memset(dis,INF,sizeof(dis));
	dis[s]=0,rnum[s]=1,pnum[s]=num[s],pre[s]=-1;
	pq.push(Edge(s,0));
	while(!pq.empty()){
		tmp=pq.top();
		pq.pop();
		u=tmp.v;
		if(vis[u]) continue;
		vis[u]=true;
		if(u==d) break;
		for(int i=0;i<e[u].size();i++){
			v=e[u][i].v,w=e[u][i].w;
			if(!vis[v]){
				if(dis[u]+w<dis[v]){
					dis[v]=dis[u]+w;
					rnum[v]=rnum[u];
					pre[v]=u;
					pnum[v]=pnum[u]+num[v];
					pq.push(Edge(v,dis[v]));
				}
				else if(dis[u]+w==dis[v]){
					rnum[v]+=rnum[u];
					if(pnum[u]+num[v]>pnum[v]){
						pnum[v]=pnum[u]+num[v];
						pre[v]=u;
					}
				}
			}
		}
	}
}
stack<int> ans;
int main(){
	cin>>n>>m>>s>>d;
	For(i,0,n-1) cin>>num[i];
	For(i,1,m) cin>>u>>v>>w,e[u].p_b(Edge(v,w)),e[v].p_b(Edge(u,w));
	dij();
	cout<<rnum[d]<<" "<<pnum[d]<<endl;
	while(d!=-1){
		ans.push(d);
		d=pre[d];
	}
	while(ans.size()>1){
		cout<<ans.top()<<" ";
		ans.pop();
	}
	cout<<ans.top();
	return 0;
}