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
解题思路
不知道是不是把不必要的东西都写在标题上了...这题我只拿了22分 依然有一个点没过 然而我比较懒就没有去看dijkstra写法了。
网上的大多数方法都是基于Dijkstra算法的,这里的spfa更加简洁,写起来也不宜出错,那让我们开始吧。
首先使用邻接矩阵去存储这个图,存储邻接矩阵有两个好的方法一个方法是 int Map[maxn][maxn];然后Map[u][v] = w;Map[v][u] = w;这个方法比较不节省内存,是原先c的写法;那么换作c++的写法呢? vector<vector<Nodetype> > vec(maxn);
在插入的时候你可以写成一个节点的形式,举例子:
const int maxn = 10010;
struct node
{
int v,w;
node(int x = 0,int y = 0):v(x),w(y){}
};
node Nodes[maxn];
int u,v,w;
cin>>u>>v>>w;
Nodes[u].push_back(node(v,w));
Nodes[v].push_back(node(u,w));
写成这样一个样子,当然如果这是一个不带权重的路径你直接写成 Node[u].push_back(v);就可以了;
然后存储完这个图之后就要spfa最短路径去求要求的东西了。
首先找出最短路径,在找最短路径的过程中如果有遇到Dis[j] == Dis[p] + Map[p][j] 也就是在进行边松弛的时候发现两个不同但是长度一样的路径,那么就应该让这个节点的值+1了,并且如果路径长度相等但是获得的增援数量更多,那么就需要改变数量以及前缀数组Pre了,这样一来既改变了路径有改变了增援数;
#include <iostream>
#include <bits/stdc++.h>
using namespace std;
const int INF = 0x3f3f3f3f;
const int MAXN = 500+10;
int teamCity[MAXN];
int _map[MAXN][MAXN];
bool Vis[MAXN];
int Dis[MAXN];
int price[MAXN];
int Rode[MAXN];
int Pre[MAXN];
int N,M,S,D;
void Spfa(int s){
for (int i = 0; i < N; ++i)
{
Dis[i] = INF;
Vis[i] = false;
price[i] = 0;
Rode[i] = 0;
Pre[i] = i;
}
Vis[s] = true;
Dis[s] = 0;
Rode[s] = 1;
price[s] = teamCity[s];
queue<int> q;
q.push(s);
while(!q.empty()){
int p = q.front();
// cout<<"p is "<<p<<endl;
q.pop();
Vis[p] = false;
for(int j = 0;j<N;j++){
if( Dis[j] > Dis[p] + _map[p][j] && _map[p][j]!= INF){
Pre[j] = p;
// cout<<p<<" to "<<j<<endl;
Dis[j] = Dis[p] + _map[p][j];
price[j] = price[p] + teamCity[j];
Rode[j] = Rode[p] ;
if(Vis[j] == false)
{
Vis[j] = true;
q.push(j);
}
}else if(Dis[j] == Dis[p] + _map[p][j] ){
if(price[p] +teamCity[j] > price[j]){
price[j] = price[p] + teamCity[j];
Pre[j] = p;
Rode[j] += Rode[p];
if(Vis[j] == false)
{
Vis[j] = true;
q.push(j);
}
}
}
}
}
}
int main(int argc, char const *argv[])
{
cin>>N>>M>>S>>D;
for (int i = 0; i < N; ++i)
{
cin>>teamCity[i];
}
for (int i = 0; i < N; ++i)
{
for (int j = 0; j < N; ++j)
{
_map[i][j] = INF;
}
}
for (int i = 0; i < M; ++i)
{
int u,v,w;
cin>>u>>v>>w;
_map[u][v] = w;
_map[v][u] = w;
}
if(S == D){
cout<<0<<" "<<teamCity[S]<<endl;
cout<<S<<endl;
return 0;
}
Spfa(S);
cout<<Rode[D]<<" "<<price[D]<<endl;
int res = D;
stack<int> Sta;
while(Pre[res]!= res){
Sta.push(res);
res = Pre[res];
}
Sta.push(res);
int flag = 1;
while(!Sta.empty()){
int f = Sta.top();
if(flag){
cout<<f;
flag = 0;
}
else
cout<<" "<<f;
Sta.pop();
}
return 0;
}