PTA 数据结构与算法题目集(中文) 7-9 旅游规划(25 分) Dijkstra 单源最短路算法

7-9 旅游规划(25 分)

有了一张自驾旅游路线图,你会知道城市间的高速公路长度、以及该公路要收取的过路费。现在需要你写一个程序,帮助前来咨询的游客找一条出发地和目的地之间的最短路径。如果有若干条路径都是最短的,那么需要输出最便宜的一条路径。

输入格式:

输入说明:输入数据的第1行给出4个正整数N、M、S、D,其中N(2≤N≤500)是城市的个数,顺便假设城市的编号为0~(N−1);M是高速公路的条数;S是出发地的城市编号;D是目的地的城市编号。随后的M行中,每行给出一条高速公路的信息,分别是:城市1、城市2、高速公路长度、收费额,中间用空格分开,数字均为整数且不超过500。输入保证解的存在。

输出格式:

在一行里输出路径的长度和收费总额,数字间以空格分隔,输出结尾不能有多余空格。

输入样例:

4 5 0 3
0 1 1 20
1 3 2 30
0 3 4 10
0 2 2 20
2 3 1 20

输出样例:

3 40

 在dijkstra算法的基础上,加了一个在路径长度相等的情况下的最小花费。只需要在松弛操作中加一条d[u]+edge[u][j]==d[j]即可。

代码如下:

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;
const int maxn=505;
const int INF=0x3f3f3f3f;
int n,m,s,d;
//是否判断在集合中
int vis[maxn];
//与源点的最短路径
int dj[maxn];
//两点之间的花费
int spend[maxn][maxn];
//与原点的最短花费
int sp[maxn];
//两点之间的距离
int edge[maxn][maxn];
//初始化函数
void init ()
{
    memset (vis,0,sizeof(vis));
    for (int i=0;i<n;i++)
       for (int j=0;j<n;j++)
           i==j? edge[i][j]=0,spend[i][j]=0:edge[i][j]=INF,spend[i][j]=INF;
}
//最短路
void djst ()
{
    for (int i=0;i<n;i++)
    {
        dj[i]=edge[s][i];
        sp[i]=spend[s][i];
    }
    while (1)
    {
       int k=-1;
       int maxx=INF;
       for (int i=0;i<n;i++)
       {
           if(maxx>dj[i]&&!vis[i])
           {
               maxx=dj[i];
               k=i;
           }
       }
       if(k==-1)
        break;
        vis[k]=1;
       for (int i=0;i<n;i++)
       {
           //松弛操作
           if(dj[k]+edge[k][i]<dj[i]&&!vis[i])
           {
               dj[i]=dj[k]+edge[k][i];
               sp[i]=sp[k]+spend[k][i];
           }
           //取最小花费
           if(dj[k]+edge[k][i]==dj[i]&&!vis[i])
           {
               sp[i]=min(sp[i],sp[k]+spend[k][i]);
           }
       }
    }
}
int main()
{
    scanf("%d%d%d%d",&n,&m,&s,&d);
    init ();
    for (int i=0;i<m;i++)
    {
        int x,y,len1,spend1;
        scanf("%d%d%d%d",&x,&y,&len1,&spend1);
        edge[x][y]=edge[y][x]=len1;
        spend[x][y]=spend[y][x]=spend1;
    }
    djst();
    printf("%d %d\n",dj[d],sp[d]);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_41410799/article/details/81809076