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
其实就算是直捣黄龙的简单版,但是数组要开够。因为是500*500条边,本来算少了一个0,不过内存空间吧,想用就用吧。浙江省赛有可能会卡,我被卡过
以下代码均有bug,就是所有的样例s均为0,所以给过了,还是pre置-1比较好
要省略vis数组的时候,用到dis[u]的时候请用pq.top().len代替,因为没有设置vis数组,可能会被更新掉,比如给定这样的边
就会出线dis[2]<pq.top().len,没用的边就会对结果产生影响。会导致最后一个样例过不去。
最短的代码
#include<bits/stdc++.h>
using namespace std;
const int N=505,INF=0x3f3f3f3f;
#define pii pair<int,int>
#define fi first
#define se second
vector<pii >G[N];
pii p;
int dis[N],val[N],pre[N],en[N],sum[N],vis[N];
void dij(int s)
{
memset(dis,INF,sizeof dis);
memset(pre,-1,sizeof pre);
priority_queue<pii,vector<pii>,greater<pii> >q;
q.push({0,s}),dis[s]=0,sum[s]=1,en[s]=val[s];
while(!q.empty())
{
p=q.top(),q.pop();
int u=p.se,v,w;
for(auto X:G[u])
{
v=X.fi,w=X.se;
if(dis[v]>p.fi+w)
{
dis[v]=p.fi+w,q.push({dis[v],v});
pre[v]=u,en[v]=en[u]+val[v],sum[v]=sum[u];
}
else if(dis[v]&#