原题链接: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
思路:
读完题目明显要求最短路,并且要记录路径,本题关键要找到快速道路最多的最短路,所以需要深度遍历 搜索每个最短路径
rt储存的是到点的上一个点的集合,均为最短路的点
ls代表的是当前的路径,ans代表的是最优的路径
代码:
#include<bits/stdc++.h>
using namespace std;
const int N = 510;
int n,m,s,d;
int ksl[N];
int f[N][N];
bool st[N];
int dist[N];
vector<int>rt[N];
void dj()
{
dist[s] = 0;
for(int i = 1; i <= n; i ++ )
{
int t = -1;
for(int j = 0; j < n; j ++ )
{
if(!st[j] && (t == -1 || dist[j] < dist[t]))
{
t = j;
}
}
st[t] = 1;
for(int j = 0; j < n; j ++ )
{
if(dist[t] + f[t][j] < dist[j])
{
dist[j] = dist[t] + f[t][j];
rt[j].clear();
rt[j].push_back(t);
}
else if(dist[t] + f[t][j] == dist[j])
{
rt[j].push_back(t);
}
}
}
}
int p = 0, k = 0; //p代表最短路径的条数, k代表救援队最大数量
vector<int>ls, ans;
void dfs(int l)
{
if(l == s) //到了起点
{
p++; //最短路径的条数
ls.push_back(l); //加入当前路径
int su = 0;
for(int i = 0; i < ls.size(); i ++ )
{
int a = ls[i];
int b = ksl[a];
su += b;
}
if(su > k) //当前方案能召集的救援队数量大于最大数量
{
k = su; //更新最大数量
ans = ls; //更新最优路径
}
ls.pop_back(); //恢复现场
}
else
{
ls.push_back(l); //加入当前路径
for(int i = 0; i < rt[l].size(); i ++ ) //遍历该点的上一个点
{
dfs(rt[l][i]);
}
ls.pop_back(); //恢复现场
}
}
int main()
{
memset(dist, 0x3f, sizeof dist);
memset(f, 0x3f, sizeof f);
cin>>n>>m>>s>>d;
for(int i = 0; i < n; i ++ ) cin>>ksl[i];
for(int i = 1; i <= m; i ++ )
{
int x,y,z;
cin>>x>>y>>z;
f[x][y] = z;
f[y][x] = z;
}
dj();
dfs(d);
cout<<p<<" "<<k<<endl;
for(int i = ans.size() - 1; i >= 0; i -- )
{
if(i != 0) cout<<ans[i]<<" ";
else cout<<ans[i];
}
return 0;
}