PTA 城市间紧急救援

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

输入格式:

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

一开始拿到题目的时候，想的是直接用搜索打印路径结果就是T最后一个点（意料之中）

代码如下：

#include<bits/stdc++.h>
using namespace std;
int h[505];
int vis1[505];
vector<int> G[505];
vector<int> v[505];
vector<int> v1,v2;
int n,m,start,last,ans1;
int minn,maxx;
void dfs(int tx,int sum,int num) {
	if(tx==last) {
		if(sum==minn) {
			ans1++;
			if(maxx<num) {
				maxx=num;
				v1=v2;
			}
		}else if(sum<minn){
			minn=sum;
			ans1=1;
			v1=v2;
			maxx=num;
		}
		return ;
	}
	for(int i=0; i<G[tx].size(); i++) {
		if(vis1[G[tx][i]]==0) {
			vis1[G[tx][i]]=1;
			v2.push_back(G[tx][i]);
			dfs(G[tx][i],sum+v[tx][i],num+h[G[tx][i]]);
			v2.pop_back();
			vis1[G[tx][i]]=0;
		}
	}
}

int main() {
	cin>>n>>m>>start>>last;
	for(int i=0; i<n; i++) {
		cin>>h[i];
	}
	for(int i=0; i<m; i++) {
		int a,b,c;
		cin>>a>>b>>c;
		G[a].push_back(b);
		G[b].push_back(a);
		v[a].push_back(c);
		v[b].push_back(c);
	}
	minn=INT_MAX;
	maxx=0,ans1=0;
	dfs(start,0,h[start]);
	cout<<ans1<<' '<<maxx<<endl;
	cout<<start;
	for(int i=0; i<v1.size(); i++) {
		cout<<' '<<v1[i];
	}
	return 0;
}

正解：这是一题最短路的算法题，通过一趟dijkstra来寻找最短路径的同时，也可以知道从哪个点可以到达终点从而达到记录路径的目的。在到达距离相同时，唯有之前救援队的数量大于当前所能取得的最大的救援队的数量时，才会更新。

代码：

#include<bits/stdc++.h>
using namespace std;
int s[505][505];
int vis[505];
int ans[505];
int h[505];
int f[505];    //记录路径 
int sum[505];  //记录每个点所能取得的最大的消防队的数量 
int n,m,start,last,ans1;
void init(){   //初始 
	for(int i=0;i<n;i++){
		for(int j=0; j<n; j++){
			s[i][j]=5000005;
		}
	}
	for(int i=0;i<n;i++){
		sum[i]=h[i];
	}
	return ;
}
int num[505];
void dijkstra(){
	vis[start]=1;
	f[start]=-1;            //起点前没有点，而本题是从0开始的，所以标记为-1 
	num[start]=1;
	for(int i=0;i<n;i++){
		ans[i]=s[start][i];
		if(ans[i]!=5000005){
			num[i]=1;
		}
	}
	ans[start]=0;
	for(int i=0;i<n-1;i++){
		int minn=5000005;
		int id=-1;
		for(int j=0;j<n;j++){
			if(!vis[j] && ans[j]<minn){
				id=j;
				minn=ans[j];
			}
		}
		vis[id]=1;
		if(id==-1){
			return ;
		}
		for(int j=0;j<n;j++){
			if(!vis[j]){
				if(ans[j]>ans[id]+s[id][j]){
					ans[j]=min(ans[j],ans[id]+s[id][j]);
					f[j]=id;
					sum[j]=h[j]+sum[id];
					num[j]=num[id];
				}else if(ans[j]==ans[id]+s[id][j]){
					if(sum[j]<h[j]+sum[id]){ 
                    //只有在当前最多救援队少于从这点点过来在加上本身的救援队数量才更改 
						f[j]=id;
						sum[j]=h[j]+sum[id];
					}
					num[j]+=num[id];        // 合并最短路径 
				}
			}
		}
	}
}

int main(){
	cin>>n>>m>>start>>last;
	for(int i=0;i<n;i++){
		cin>>h[i];
	}
	memset(vis,0,sizeof vis);
	init();
	for(int i=0;i<m;i++){
		int a,b,c;
		cin>>a>>b>>c;
		if(c<s[a][b]){
			s[a][b]=s[b][a]=c;
		}
	}

	dijkstra();
	vector<int> v;
	int cnt=last;
	while(f[cnt]!=-1){
		v.push_back(f[cnt]);
		cnt=f[cnt];
	}
	cout<<num[last]<<' '<<sum[last]+h[start]<<endl;
	for(int i=v.size()-1;i>=0;i--){
		cout<<v[i]<<' ';
	}
	cout<<last<<endl;
	return 0;
}