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
好久没有敲代码,不仅是手生,脑瓜也没有之前灵活了,以后还是应该每天保持敲代码!
想着好久没有写代码,今天就来复习了之前最简单的知识点---最短路径
这一个代码改了好久,敲代码的好习惯不能丢,每天坚持做题,锻炼思维,练习算法。
AC代码
#include<bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
int n,m,s,d;
int g[1000][1000];
int people[1000];
int dis[1000],num[1000],maxpeople[1000];
int path[1000],vis[1000],pre[1000];
void Dijkstra(int s)
{
for(int i=0;i<n;i++)
{
vis[i]=0;
maxpeople[i]=0;
num[i]=0;
dis[i]=INF;
}
memset(dis,INF,sizeof(dis));
dis[s]=0;
num[s]=1;
maxpeople[s]=people[s];
for(int i=0;i<n-1;i++)
{
int u=-1,t=INF;
for(int j=0;j<n;j++)
{
if(vis[j]==0 && t>dis[j])
{
u=j;
t=dis[j];
}
}
if(u==-1)
{
break;
}
vis[u]=1;
for(int v=0;v<n;v++)
{
if(vis[v]==0 && g[u][v]!=INF)
{
if(dis[v]>dis[u]+g[u][v])
{
dis[v]=dis[u]+g[u][v];
num[v]=num[u];
maxpeople[v]=maxpeople[u]+people[v];
pre[v]=u;
}
else if(dis[v]==dis[u]+g[u][v])
{
num[v]=num[v]+num[u];
if(maxpeople[v]<maxpeople[u]+people[v])
{
maxpeople[v]=maxpeople[u]+people[v];
pre[v]=u;
}
}
}
}
}
}
int main()
{
scanf("%d%d%d%d",&n,&m,&s,&d);
for(int i=0;i<n;i++)
{
scanf("%d",&people[i]);
}
memset(g,INF,sizeof(g));
for(int i=0;i<m;i++)
{
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
g[x][y]=g[y][x]=min(g[x][y],z);
}
Dijkstra(s);
int i=0;
int res=d;
while(res!=s)
{
path[i++]=res;
res=pre[res];
}
printf("%d %d\n",num[d],maxpeople[d]);
path[i]=s;
for(int j=i;j>0;j--)
{
printf("%d ",path[j]);
}
printf("%d\n",d);
return 0;
}