L2-001 紧急救援(无向图+有权+dijkstra算法+找前驱)
作为一个城市的应急救援队伍的负责人,你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候,你的任务是带领你的救援队尽快赶往事发地,同时,一路上召集尽可能多的救援队。
输入格式:
输入第一行给出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算法的精髓。
二。记录好前驱,前驱最终是唯一的。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
const int inf=0x3f3f3f3f;
int N, M, S, D;
int map[550][550];
int val[550], dis[550], vis[550], cnt[550], sum[550], pre[550];
//每点权值 距S距离,线段的权 标记 路径数量 最大救援队 前驱
void path(int d)
{
if(pre[d] != -1)
{
path(pre[d]);
printf("%d ", pre[d]);
}
}
int main()
{
scanf("%d %d %d %d", &N, &M, &S, &D);
int i,j;
for(i=0;i<N;i++)
scanf("%d",&val[i]);
for(i=0;i<N;i++)
{
for(j=0;j<N;j++)
{
if(i==j)
map[i][j]=0;
else map[i][j]=inf;
}
} //无向图的构造1 map[x][y]==y-->x的距离,
int x, y, z;
for(i=0;i<M;i++)
{
scanf("%d %d %d", &x, &y, &z);
map[x][y]=z;
map[y][x]=z; //无向图的构造2.
}
for(i=0;i<N;i++) //初始化
{
dis[i] = inf;
vis[i] = 0;
cnt[i] = 0;
sum[i] = 0;
pre[i] = -1;
}
dis[S] = 0; //初始化起始点
vis[S] = 1;
cnt[S] = 1; //起点到其本身方式为1
sum[S] = val[S];
for(i=0;i<N;i++)
{
int minn = inf, mini = S; //minn在新的开始又变成了无穷大
for(j = 0; j < N; j++)
{
if(vis[j]==0&&minn>dis[j])
{
minn = dis[j];
mini = j;
}
}
vis[mini]=1; //做标记,dis[mini]记录的就是到起点的最短距离,同一级别位置的
for(j=0;j<N;j++)
{
if(vis[j]==0) //vis做标记是层进关系的,前面的就不在考虑了
{
if(dis[j]>dis[mini]+map[mini][j]) //这里就有层进节点了
{
dis[j] = dis[mini] + map[mini][j];
sum[j] = sum[mini] + val[j]; //救援队肯定是包括自己救援队的
cnt[j] = cnt[mini]; //方式等于到达前驱的方式
pre[j] = mini; //记下其前驱
}
else if(dis[j]==dis[mini]+map[mini][j])
{
cnt[j]=cnt[mini]+cnt[j]; //过你的条数+不过你的条数
if(sum[j] < sum[mini] + val[j])
{ //能获得更大的救援,则更新
sum[j] = sum[mini] + val[j];
pre[j] = mini;
}
}
}
}
}
printf("%d %d\n", cnt[D], sum[D]);
path(D);
printf("%d\n", D);
return 0;
}