思路
因为project3-normal是一道差不多的题,这题更加简单,于是先做这题找思路hh
本质上是一道模板题,修改一下经典dijkstra算法即可
在更新路径长度的时候进行判断,同时更新路径数量和点权值
这是mooc上的ppt,但个人觉得有点小错误,对于等长路不能直接简单相加,需要判断边数大小再选择是否更新。
题目
作为一个城市的应急救援队伍的负责人,你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候,你的任务是带领你的救援队尽快赶往事发地,同时,一路上召集尽可能多的救援队。
输入格式:
输入第一行给出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
结尾无空行
代码(AC)
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=510;
int n,m,s,d;
//n个城市,m条道路,s-起点,d-终点
int graph[N][N];
int w[N];//权值-救援队数量
int dist[N],cnt[N],num[N],pre[N];
bool st[N];
void dij(){
memset(dist,0x3f,sizeof dist);
dist[s]=0;
cnt[s]=1;
num[s]=w[s];
for(int i=0;i<n;i++){
int v=-1;
for(int j=0;j<n;j++){
if(!st[j] && (v==-1 || dist[j]<dist[v]))v=j;
}
if(v==-1)break;
st[v]=true;
for(int u=0;u<n;u++){
if(!st[u]){
if(dist[u]>dist[v]+graph[u][v]){
dist[u]=dist[v]+graph[u][v];
cnt[u]=cnt[v];
num[u]=w[u]+num[v];
pre[u]=v;
}else if(dist[u]==dist[v]+graph[u][v]){
cnt[u]+=cnt[v];
if(w[u]+num[v]>num[u]){
num[u]=w[u]+num[v];
pre[u]=v;
}
}
}
}
}
}
void printpath(int x){
if(x==s){
printf("%d ",x);
return;
}
printpath(pre[x]);
printf("%d",x);
if(x!=d)printf(" ");
}
int main(){
cin>>n>>m>>s>>d;
for(int i=0;i<n;i++)scanf("%d",&w[i]);
memset(graph,0x3f,sizeof graph);
memset(pre,-1,sizeof pre);
while(m--){
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
if(c<graph[a][b])graph[a][b]=c;
if(c<graph[b][a])graph[b][a]=c;
}
dij();
printf("%d %d\n",cnt[d],num[d]);
printpath(d);
return 0;
}
顺便一提,不知道为什么最近编译器好像开启了自动纠错功能,debug时间大大减少(orz