作为一个城市的应急救援队伍的负责人,你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候,你的任务是带领你的救援队尽快赶往事发地,同时,一路上召集尽可能多的救援队。
输入格式:
输入第一行给出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
解题思路:这是一个很明显的最短路径问题,但是要考虑到路径相同的情况下,要求要有更多的救援队的数量。因此,和常规的迪杰斯特拉算法相比,要增加一个if的判断条件,用来判断上述情况。下面是我改造过的大神的代码,我给出了许多的注释,不对的地方,希望大家一起讨论!
#include <bits/stdc++.h>
#define MAXN 10000
using namespace std;
int dis[MAXN],weight[MAXN],e[MAXN][MAXN],num[MAXN],w[MAXN],pre[MAXN];
bool visit[MAXN];
const int inf=1000000;
int n,m,start,finish;
//输出路径
void print(int v)
{
if(v==start)
{
printf("%d",v);
return;
}
print(pre[v]);
printf(" %d",v);
}
int main()
{
//n个城市,m个快速车道的个数,start是起点,finish是终点
scanf("%d %d %d %d",&n,&m,&start,&finish);
for(int i=0;i<n;i++)
{
scanf("%d",&weight[i]);//每个城市的救援队数量
}
//把所有的值均置为inf
fill(e[0],e[0]+MAXN*MAXN,inf);
fill(dis,dis+MAXN,inf);
int a,b,c;
for(int i=0;i<m;i++)
{
scanf("%d %d %d",&a,&b,&c);
//构建图,双向边
e[a][b]=c;
e[b][a]=c;
}
//起点初始化
dis[start]=0;
w[start]=weight[start];//存入c1救援队的数量
num[start]=1;//路过城市的数量
for(int i=0;i<n;i++)
{
//把构建的图中的顶点均加入到dis数组中
//此时的dis数组中的数值还都是inf
//为了得到当前的最小路径,在未获得最短路径的顶点在找到离c1最近的顶点
int u=-1,minn=inf;
for(int j=0;j<n;j++)
{
//若该点没有被访问,就把该顶点加入到dis数组中
if(visit[j]==false&&dis[j]<minn)
{
u=j;
minn=dis[j];
}
}
if(u==-1)
break;
//标记u已经取得最短路径
visit[u]=true;
for(int v=0;v<n;v++)
{
//假设该顶点没有被访问,并且边的长度不是最大值
if(visit[v]==false&&e[u][v]!=inf)
{
//已经存在的最短路径+边的长度<该点的路径长度
if(dis[u]+e[u][v]<dis[v])
{
dis[v]=dis[u]+e[u][v];//加入到dis数组中
num[v]=num[u];//更新计数(城市)数组
w[v]=w[u]+weight[v];//更新救援队的数量
pre[v]=u;//把已存在的最短路径的节点加入到pre数组
}
//假设已存在最短路径长度+边的长度=当前点的长度
//根据本题的特点需要再次考虑城市救援队的数量,把数量大的城市顶点加入
else if(dis[u]+e[u][v]==dis[v])
{
num[v]=num[v]+num[u];
if(w[u]+weight[v]>w[v])
{
w[v]=w[u]+weight[v];
pre[v]=u;
}
}
}
}
}
//输出截止到终点城市,经过的城市数目和救援队的数量
printf("%d %d\n",num[finish],w[finish]);
print(finish);//输出最短路径的长度
return 0;
}