L2-001 紧急救援
https://pintia.cn/problem-sets/994805046380707840/problems/994805073643683840
题意
作为一个城市的应急救援队伍的负责人,你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候,你的任务是带领你的救援队尽快赶往事发地,同时,一路上召集尽可能多的救援队。
输入
输入第一行给出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算法,求最短路,当最短路不唯一时,选择所经过城市的救援人数总数最多的一条。road[i]=j表示到达城市 i 的最短路径有 j 条数,具体看程序它是如何计算的(也可改用dfs求解最短路的条数)。
C++程序
#include<iostream>
#include<vector>
#include<queue>
using namespace std;
const int N=505;
const int INF=0x3f3f3f3f;
//边
struct Edge{
int v,cost;
Edge(int v,int cost):v(v),cost(cost){}
};
//结点
struct Node{
int u,cost;
Node(){}
Node(int u,int cost):u(u),cost(cost){}
bool operator <(const Node &q)const//使用优先队列,因此要重载<运算符
{
return cost>q.cost;
}
};
vector<Edge>g[N];//图的邻接矩阵
bool vis[N];
int n,path[N],dist[N],tmp_num[N],num[N],road[N];//road[i]表示到达结点i的路径数
//打印路径
bool flag=false;
void print(int e)
{
if(path[e]!=-1) print(path[e]);
if(flag) printf(" ");
flag=true;
printf("%d",e);
}
//最短路算法
void dijkstra(int s,int e)
{
for(int i=0;i<n;i++)//初始化
{
vis[i]=false;
road[i]=0;
path[i]=-1;
dist[i]=INF;
tmp_num[i]=0;
}
priority_queue<Node>q;
q.push(Node(s,0));
dist[s]=0,tmp_num[s]=num[s];
road[s]=1;//初始化到达城市s的最短路条数为1
while(!q.empty())
{
Node f=q.top();
q.pop();
int u=f.u;
if(!vis[u])
{
vis[u]=true;
for(int i=0;i<g[u].size();i++)
{
int v=g[u][i].v;
int len=dist[u]+g[u][i].cost;
int cnt=tmp_num[u]+num[v];
if(len<dist[v])
road[v]=road[u];//此时从u走到v,v的最短路条数和u一样
else if(len==dist[v])
road[v]+=road[u];//此时可以选择从u走到v,v的最短路条数要加上这额外的
if(len<dist[v]||(len==dist[v]&&cnt>tmp_num[v]))//满足条件就进行更新
{
path[v]=u;
dist[v]=len;
tmp_num[v]=cnt;
q.push(Node(v,dist[v]));
}
}
}
}
printf("%d %d\n",road[e],tmp_num[e]);
print(e);
printf("\n");
}
int main()
{
int m,s,d;
scanf("%d%d%d%d",&n,&m,&s,&d);
for(int i=0;i<n;i++)
scanf("%d",&num[i]);
for(int i=1;i<=m;i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
g[a].push_back(Edge(b,c));
g[b].push_back(Edge(a,c));
}
dijkstra(s,d);
return 0;
}