作为一个城市的应急救援队伍的负责人,你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候,你的任务是带领你的救援队尽快赶往事发地,同时,一路上召集尽可能多的救援队。
输入格式:
输入第一行给出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算法,固定套路+根据题意添加变量做修改。
只是数据量稍微多些,我把数据量分为了三类。
1.输入 2. 输出3. 固定套路数据量
记住算法中固定套路的数据量比如d[maxn] ,vis[maxn],w[maxn]
等再根据题意添加数据量这样会简单些。
具体思路:
首先分析题目明确目的,要求输出最短路径条数,还要能够召集最多的救援队数量,还有输出经过的城市编号。
接下来开始求解。从起点,找距离最近的城市点u,找到后标记u已被访问,确定u作为中转站,通过d[v] > d[u] + G[u][v]
更新dv;反复重复n次,直到所有城市点都被标记。这就是一般dijskra大致思路再结合目的细化考虑。
最短路径条数,召集最多救援队数量,记录前结点都应该伴随在更新dv的步骤后。更新dv条件:d[v] <d[u] + G[u][v] || (d[v] == d[u] + G[u][v] && w[v] < w[u] + weight[v]
最后,深度遍历输出
注意: 要记得d[maxn] 、G[maxn][maxn]初始化为INF
#include <iostream>
#include <cstdio>
#include <memory>
#include <algorithm>
using namespace std;
const int maxn = 550;
const int INF = 0x3fffffff;
int n,m,s,e;
int G[maxn][maxn], weight[maxn];
int w[maxn],d[maxn],pre[maxn],num[maxn];
bool visit[maxn] = {false};
void dfs(int v)
{
if(v == s)
{
cout << s;
return;
}
else
{
dfs(pre[v]);
cout << " " << v;
}
}
void dijkstra(int s)
{
fill(d, d + maxn, INF);
fill(w, w + maxn, 0);
fill(num, num + maxn, 0);
d[s] = 0;
w[s] = weight[s];
num[s] = 1;
for(int i = 0; i < n; i++)
{
int u = -1, MIN = INF;
for(int j = 0; j < n; j++)
{
if(visit[j] == false && d[j] < MIN)
{
u = j;
MIN = d[u];
}
}
if(u == -1) return;
visit[u] = true;
for(int v = 0; v < n; v++)
{
if(visit[v] == false && G[u][v] != INF)
{
if(d[u] + G[u][v] < d[v])
{
d[v] = d[u] + G[u][v];
w[v] = w[u] + weight[v];
num[v] = num[u];
pre[v] = u;
}
else if(d[u] + G[u][v] == d[v])
{
if(w[v] < w[u] + weight[v])
{
w[v] = w[u] + weight[v];
pre[v] = u;
}
num[v] += num[u];
}
}
}
}
}
int main()
{
fill(G[0],G[0]+maxn * maxn, INF); //都是不可达的
cin >> n >> m >> s >> e;
for(int i = 0; i < n; i++)
{
cin >> weight[i];
}
while(m--)
{
int u,v;
cin >> u >> v;
cin >> G[u][v];
G[v][u] = G[u][v];
}
dijkstra(s);
cout << num[e] << " " << w[e] << endl;
dfs(e);
return 0;
}