pat 城市救援最短路

最新推荐文章于 2025-04-28 20:01:17 发布

原创最新推荐文章于 2025-04-28 20:01:17 发布 · 1k 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#最短路 PAT

acm 同时被 3 个专栏收录

188 篇文章

订阅专栏

codeproblem

106 篇文章

订阅专栏

studyproblem

96 篇文章

订阅专栏

本文介绍了一道结合最短路径问题和点权和最大值的题目，涉及城市应急救援场景。通过迪杰斯特拉算法解决最短路问题，并在路径中寻找能召集的最多救援队。代码示例使用SPFA方法实现，强调在路径更新时考虑点权和最大值的优化。

<span style="font-family: Arial, Helvetica, sans-serif; background-color: rgb(255, 255, 255);"></span><pre name="code" class="cpp"><span style="font-family: Arial, Helvetica, sans-serif; background-color: rgb(255, 255, 255);">5-12 城市间紧急救援   (25分)</span>

作为一个城市的应急救援队伍的负责人，你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候，你的任务是带领你的救援队尽快赶往事发地，同时，一路上召集尽可能多的救援队。

输入格式:

输入第一行给出4个正整数 $N$ 、 $M$ 、 $S$ 、 $D$ ，其中 $N$ （ $2\le N\le 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

这是一道最短路基础上求最短路径长度和点权和最大值的问题,同时要记录最短路径.

运用迪杰斯特算法求出最短路,在做松弛时,由于要记录最短路径长度,所以当dis[i]=dis[j]+map[i][j]时,说明存在另外的路径让当前点到起点的距离最小,需要把pathnum[j](到达j点的最短路径长度,下同)加到pathnum[i]上.

用toval数组记录当前点满足最短路前提下的点权和的最大值,每次松弛操作都要更新,当dis[i]=dis[j]+map[i][j]时,判断total[j]+b[j](当前点的权值)是否大于total[i],如果大于就需要更新. 用pre数组作前缀数组,在每次松弛操作和dis[i]=dis[j]+map[i][j]相等时更新.

代码:

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
int n, m, s,q;
int b[600];
int a[600][600];
int c[600];
int pre[600];
int inf;
int sx[600];
int toval[600];
int main()
{
    memset(a, 0, sizeof(0));
    int pathnum[600];
    inf=99999999;
    scanf("%d%d%d%d", &n, &m, &s, &q);
    int i;
    int j;
    for(i=0; i<n; i++)
        for(j=0; j<n; j++)
        {
            if(i==j)a[i][j]=0;
            else a[i][j]=inf;
        }
    for(i=0; i<n; i++)scanf("%d", &b[i]);
    for(i=0; i<m; i++)
    {
        int x, y, z;
        scanf("%d%d%d", &x, &y, &z);
        if(z<a[x][y])  a[x][y]=a[y][x]=z;
    }
    int vis[600];
    for(i=0; i<n; i++)
    {
        vis[i]=0;
        c[i]=inf;
        toval[i]=0;
    }
    int k;
    c[s]=0;
    vis[s]=1;
    pathnum[s]=1;
    toval[s]=b[s];
    {
        for(i=0; i<n; i++)
        {
            int mini, minn;
            mini=s;
            minn=inf;
            for(j=0; j<n; j++)
            {

                if(vis[j]==0 && minn>c[j])
                {
                    mini=j;
                    minn=c[j];
                }
            }
            vis[mini]=1;
            for(j=0; j<n; j++)
            {
                if(vis[j]==0)   if(c[mini]+a[mini][j]<c[j])
                    {
                        c[j]=c[mini]+a[mini][j];
                        pre[j]=mini;
                        toval[j]=b[j]+toval[mini];
                        pathnum[j]=pathnum[mini];
                    }
                    else if(c[mini]+a[mini][j]==c[j] )
                    {
                        if( toval[j]<b[j]+toval[mini])
                        {
                            toval[j]=b[j]+toval[mini];
                            pre[j]=mini;
                        }
                        pathnum[j]+=pathnum[mini];
                    }
            }
        }
    }
    int top=0;
    j=q;
    while(j!=s)
    {
        sx[top++]=j;
        j=pre[j];
    }
    sx[top]=j;
    printf("%d %d\n", pathnum[q], toval[q]);
    for(; top>=0; top--)
    {
        printf("%d", sx[top]);
        if(top!=0)printf(" ");
    }
    return 0;
}

spfa做法：

tot数组记录到达这个点的城市数量，注意当dis[x]+w==dis[y]的时候，tot[y]+=tot[x]，而不是tot[y]++

#include <bits/stdc++.h>
#define ps push_back
using namespace std;
struct node{
    int to;
    int w;
};
const int inf=1e9+7;
vector<node>edg[505];
int dis[505], st[505], n;
int num[505];
int from[505];
int tot[505];
int book[505];
int que[10000005];
int m, s, d;
void init()
{
    for(int i=0; i<n; i++)
    {
        dis[i]=inf;
        st[i]=0;
        from[i]=-inf;
    }
}
void spfa()
{
    int i, j, head=0, tail=0;
    que[tail++]=s;
    book[s]=1;
    while(head<tail)
    {
        int x=que[head];
        for(i=0; i<(int)edg[x].size(); i++)
        {
            int y=edg[x][i].to;
            int w=edg[x][i].w;
            if(dis[x]+w<dis[y])
            {
                dis[y]=dis[x]+w;
                st[y]=num[y]+st[x];
                tot[y]=1;
                from[y]=x;
                if(book[y]==0)
                {
                    book[y]=1;
                    que[tail++]=y;
                }
            }
            else if(dis[x]+w==dis[y])
            {
                
                if(st[y]<num[y]+st[x])
                {
                    tot[y]+=tot[x];
                    st[y]=num[y]+st[x];
                    from[y]=x;
                    if(book[y]==0)
                    {
                        book[y]=1;
                        que[tail++]=y;
                    }
                }
            }
        }
        book[x]=0;
        head++;
    }
    
    return;
}
int path[505], top;
int qqq[1000][1000];
int main()
{

    cin>>n>>m>>s>>d;
    for(int i=0; i<n; i++)
    {
        scanf("%d", &num[i]);
    }
    init();
    node tmp;
    while(m--)
    {
        int x, y, z;
        scanf("%d%d%d", &x, &y, &z);
        tmp.to=y, tmp.w=z;
        edg[x].ps(tmp);
        tmp.to=x;
        edg[y].ps(tmp);
    }
    dis[s]=0;
    st[s]=num[s];
    spfa();
    int s=0;

    int x=d;
    while(x!=-inf)
    {
        s+=num[x];
        path[top++]=x;
        x=from[x];
    }
   // if(tot[d]<5)while(1);
     printf("%d %d\n", tot[d], st[d]);
    for(int i=top-1; i>=0; i--)printf(i==0?"%d\n":"%d ", path[i]);
    return 0;
}