【PTA】L2-001 紧急救援

【PTA】L2-001 紧急救援

作为一个城市的应急救援队伍的负责人，你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候，你的任务是带领你的救援队尽快赶往事发地，同时，一路上召集尽可能多的救援队
输入格式
输入第一行给出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算法和相应模板，在此基础上增加内容

#include<bits/stdc++.h>
using namespace std;
const int maxn=510;
int n,m,c1,c2,a,b,c;
int weight[maxn],num[maxn],dist[maxn],e[maxn][maxn],pre[maxn],w[maxn];
bool st[maxn];
void printPath(int v)
{
	if(v==c1)	{printf("%d",v);return ;}
	printPath(pre[v]);
	printf(" %d",v);
	return ;
}
int main()
{
	cin>>n>>m>>c1>>c2;
	memset(dist,8,sizeof dist);	memset(e,8,sizeof e);
	for(int i=0;i<n;i++)	cin>>weight[i];
	for(int i=0;i<m;i++)
	{
		cin>>a>>b>>c;
		e[a][b]=min(e[a][b],c); //储存路径 
	}
	dist[c1]=0;	w[c1]=weight[c1];	num[c1]=1;
	for(int i=0;i<n-1;i++)
	{
		int u=-1;
		for(int j=0;j<n;j++)
			if(!st[j]&&(u==-1||dist[j]<dist[u]))
				u=j;
		st[u]=true; //判断源点 
		for(int j=0;j<n;j++)
			if(dist[u]+e[u][j]<dist[j]) //如果路径更短 则更新 
			{
				dist[j]=dist[u]+e[u][j]; //路径长度更新 
				w[j]=w[u]+weight[j]; //路径救援队总数更新 因为先确保的是路径最短 
				num[j]=num[u]; //到达该点最短路径条数更新 
				pre[j]=u;  
			}
			else if(dist[u]+e[u][j]==dist[j]) //如果相等 则加上该情况 
			{
				num[j]+=num[u]; //源点到该位置的所有最短路径加上本身具有的最短路径 
				if(w[u]+weight[j]>w[j]) //更新救援队数量 
				{
					w[j]=w[u]+weight[j];
					pre[j]=u; //pre数组里存放的是最短路径同时最大救援队数量 
				}
			}
	}
	printf("%d %d\n",num[c2],w[c2]);
	printPath(c2);
}

有不明白的可以留下评论我会更新题解思路图
希望可以尽快AC这个题