PTA L2-001 紧急救援

最新推荐文章于 2022-03-21 20:02:10 发布

原创

最新推荐文章于 2022-03-21 20:02:10 发布 · 760 阅读

0 ·

CC 4.0 BY-SA版权

该博客详细介绍了PTA L2-001紧急救援问题的背景及解决方案。内容涉及城市应急救援地图，城市与道路的数据结构，输入输出格式，以及算法实现的注意事项。博主分析了直捣黄龙的简单策略，并提醒在处理500x500规模的问题时可能遇到的内存挑战。文章提到了代码中可能出现的bug，特别是在没有使用vis数组的情况下，如何避免无效边影响结果。最后，博主分享了不同优化策略，包括堆优化的Dijkstra算法和SPFA算法，讨论了它们的时间复杂度和效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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]==p.fi+w)
            {
                if(en[v]<en[u]+val[v])pre[v