最短路PTA(2)

最新推荐文章于 2023-10-12 16:18:11 发布

蒟蒻CCA

最新推荐文章于 2023-10-12 16:18:11 发布

阅读量261

点赞数

分类专栏： acm

本文链接：https://blog.csdn.net/qq_36778821/article/details/78608325

版权

acm 专栏收录该内容

26 篇文章 0 订阅

订阅专栏

本文介绍了如何运用迪杰斯特拉算法解决一个城市应急救援队伍的最短路径问题。在一张包含多个城市和快速道路的地图中，需要从起点S到达终点D，同时集结最多的救援队。输入包括城市数量、道路数量、起点和终点，以及各城市救援队数量和道路长度。输出是最短路径的条数和救援队数量，以及路径上的城市编号。提供的代码示例展示了算法的具体应用。

摘要由CSDN通过智能技术生成

题目链接：点击打开链接

题目大意：

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

输入格式:

输入第一行给出4个正整数N、M、S、D，其中N（2）是城市的个数，顺便假设城市的编号为0 ~ (；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

代码：（迪杰斯特拉实现）

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <set>
#include<malloc.h>
#include <map>
#include <vector>
#include <stack>
#include <queue>
#define ri(n) scanf("%d",&n)
#define oi(n) printf("%d\n",n)
#define rl(n) scanf("%lld",&n)
#define ol(n) printf("%lld\n",n)
#define rep(i,l,r) for(i=l;i<=r;i++)
#define rep1(i,l,r) for(i=l;i<r;i++)
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int epg=10-8;
const int maxn=1000;
int n,m,S,D;//顶点数，边数
int e[maxn][maxn];
int vis[maxn],d[maxn],pathcount[maxn],path[maxn],ren[maxn],acount[maxn],v[maxn];
void dijiesi(int s,int t)
{
    for(int i=0; i<maxn; i++)
    {
        d[i]=inf;
        vis[i]=0;
        pathcount[i]=0;
        path[i]=-1;
        acount[i]=ren[i];
    }
    d[s]=0;
   // vis[s]=1;
    pathcount[s]=1;
    while(1)
    {
        int v=-1;
        for(int i=0; i<n; i++)
        {
            if(!vis[i]&&(v==-1||d[i]<d[v]))
                v=i;
        }
        if(v==-1)
            break;
        vis[v]=1;
        for(int i=0; i<n; i++)
        {
            if(!vis[i])
            {
                if(d[i]>d[v]+e[v][i])
                {
                    d[i]=d[v]+e[v][i];
                    path[i]=v;
                    acount[i]=acount[v]+ren[i];
                    pathcount[i]=pathcount[v];
                }
                else if(d[i]==d[v]+e[v][i])
                {
                    pathcount[i]+=pathcount[v];
                    if(acount[i]<acount[v]+ren[i])
                    {
                        acount[i]=acount[v]+ren[i];
                        path[i]=v;
                    }
                }
            }
            //d[i]=min(d[i],d[v]+e[v][i]);
        }
    }
    // return d[t];
}
int main()
{
    while(scanf("%d%d%d%d",&n,&m,&S,&D)==4)
    {
        memset(e,inf,sizeof(e));
        memset(vis,0,sizeof(vis));
        memset(d,inf,sizeof(d));
        memset(v,0,sizeof(v));
        for(int i=0; i<n; i++)
            scanf("%d",&ren[i]);
        for(int i=0; i<n; i++)
            e[i][i]=0;
        for(int i=1; i<=m; i++)
        {
            int u,v,cost;
            scanf("%d%d%d",&u,&v,&cost);
            if(e[u][v]>cost)
                //e[u][v]=cost;
                e[u][v]=e[v][u]=cost;//无向图
        }
        dijiesi(S,D);
        printf("%d %d\n",pathcount[D],acount[D]);
        int sum=0;
        int ans=D;
        while(path[D]!=S)
        {
            //cout<<path[D]<<endl;
            v[sum++]=path[D];
            D=path[D];
        }
       // v[sum++]=D;
        for(int i=sum;i>=0;i--)
        {
            if(i==sum)
                printf("%d",v[i]);
            else
                printf(" %d",v[i]);
        }
        printf(" %d\n",ans);
    }
    return 0;
}