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
code:
#include<bits/stdc++.h>
typedef long long ll;
using namespace std;
int n,m,s,d;
int weight[510],city[510][510],visit[510],pre[510],dis[510],w[510],road[510];
const int inf=99999;
void printPath(int v)//利用递归 打印路径(路径本身顺序是倒着的)
{
if(v==s)
{
printf("%d",v);//递归退出条件
return ;
}
printPath(pre[v]);
printf(" %d",v);
}
int main(){
cin>>n>>m>>s>>d;
for(int i=0;i<n;i++) cin>>weight[i];
fill(city[0],city[0]+510*510,inf);//初始化城市间距离无限
fill(dis,dis+510,inf);
for(int i=0;i<m;i++){
int a,b,c;
cin>>a>>b>>c;
city[a][b]=c;
city[b][a]=c;
}
dis[s]=0;//起点到起点距离为0
w[s]=weight[s];//能召集的队伍数
road[s]=1;
for(int i=0;i<n;i++){//一共n轮
int mincity=-1;
int mindis=inf;
for(int j=0;j<n;j++){
if(visit[j]==0&&dis[j]<mindis){
mincity=j;
mindis=dis[j];
}//找到当前可到达的最短距离的城市
}
if(mincity==-1) break;//所有城市的最短距离都找到了
visit[mincity]=1;//该城市已找到最短距离
for(int v=0;v<n;v++ ){//更新信息
if(visit[v]==0&&city[mincity][v]!=inf){//可达 且未找到最短距离
if(dis[mincity]+city[mincity][v]<dis[v]){//通过mincity到达v更短
dis[v]=dis[mincity]+city[mincity][v];//更新距离
pre[v]=mincity;
w[v]=w[mincity]+weight[v];
road[v]=road[mincity];
}else if(dis[mincity]+city[mincity][v]==dis[v]){
road[v]+=road[mincity];
if(w[mincity]+weight[v]>w[v]){//如果这条路有更多救援队 就走这条
w[v]=w[mincity]+weight[v];
pre[v]=mincity;
}
}
}
}
}
// for(int i=0;i<n;i++){
// cout<<road[i]<<" ";
// }
// cout<<endl;
// for(int i=0;i<n;i++){
// cout<<dis[i]<<" ";
// }
printf("%d %d\n",road[d],w[d]);
printPath(d);
}
dijkstra老是忘记代码怎么写。
- 初始化起点到起点距离为0
- 找到当前可达到的点中距离最小的点
- 标记该点已找到最短距离,对于其它点,若通过该点到达距离更短则更新距离。
- 循环2,3直到所有点都找到最小距离
值得注意的是,只能用fill来初始化距离inf,而memset只能初始化-1,0。