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
直接领接矩阵也能过,不过我为了复习一下堆优化dij 就写了遍
不过有个地方我用
memset()就出现段错误 不知道为啥。。。求大佬告知
//大根堆优化dij 复杂度O(m*logn)
#include<bits/stdc++.h>
using namespace std;
const int M = 1000000+100;
priority_queue<pair<int, int>, vector<pair<int, int> >, greater<pair<int, int> > >q;
int n,m,cnt,S,D;
int d[M];
int head[M];
bool v[M];
int num[M];//记录最短路径条数
int ctp[M];//记录每个城市救援人数
int mctp[M];//记录最短路最多救援人数
int pre[M];//表示最后一次更新最短路径i时,最短路径i的上一个点
//记录路径,倒序储存。用DFS可正序输出
struct EDGE
{
int to,next,w;
}edge[M];
void add(int x,int y,int w)
{
edge[++cnt].to=y;
edge[cnt].w=w;
edge[cnt].next=head[x];
head[x]=cnt;
}
void dijkstra(int origin)
{
for(int i=0;i<=n;i++)
d[i]=0x3f3f3f3f,v[i]=false,num[i]=0;
// memset(d,0x3f,sizeof(d));这里用memset会出现段错误
// memset(v,false,sizeof(v));
// memset(num,0,sizeof(num));
num[origin]=1;
//起点最短路径条数初始化为1
d[origin]=0;
q.push(make_pair(0,origin));
mctp[origin]=ctp[origin];
// printf("%d=====%d =====%d\n",mctp[origin],origin,ctp[origin]);
while(q.size()!=0)
{
int x=q.top().second;q.pop();
if(v[x])continue;
v[x]=true;
for(int i=head[x];i;i=edge[i].next)
{
int v=edge[i].to,w=edge[i].w;
if(d[v]>d[x]+w)
{
d[v]=d[x]+w;
num[v]=num[x]; //这种情况说明有更短的路径,所以路径总数不变。
q.push(make_pair(d[v],v));
pre[v]=x;
mctp[v]=mctp[x]+ctp[v];
}
else if(d[v]==d[x]+w)
{
num[v]+=num[x];//路径一样,加上之前的路径
if(mctp[v]<mctp[x]+ctp[v])
{
mctp[v]=mctp[x]+ctp[v];
pre[v]=x;
}
}
// printf("%d %d %d mctpx= %d ctpv %d\n",mctp[v],x,v,mctp[x],ctp[v]);
}
}
}
/*
2 1 0 1
1 2
0 1 1
*/
void dfs(int x)
{
if(x==S)
{
printf("%d",S);
return ;
}
dfs(pre[x]);
printf(" %d",x);
}
/*
void dijkstra(int origin)
{
memset(d,0x3f,sizeof(d));
memset(v,0,sizeof(v));
d[origin]=0;
q.push(make_pair(0,1));
while(q.size())
{
int x=q.top().second;q.pop();
if(v[x])continue;
v[x]=1;
for(int i=head[x];i;i=edge[i].next)
{
int w=edge[i].val;
int v=edge[i].to;
if(d[v] > d[x] + w)
{
d[v] = d[x] + w;
q.push(make_pair(d[v],v));
}
}
}
}*/
int main()
{
int u,v,w;
cin>>n>>m>>S>>D;
for(int i=0;i<n;i++)
cin>>ctp[i];
for(int i = 1; i<= m; i++)
{
cin>>u>>v>>w;
add(u,v,w);
add(v,u,w);
}
dijkstra(S);
/* for(int i = 1; i <= n; i++)
{
cout<<d[i]<<endl;
}*/
printf("%d %d\n",num[D],mctp[D]);
dfs(D);
puts("");
return 0;
}