PTA 07-图6 旅游规划

PTA 07-图6 旅游规划

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

输入格式:
输入说明：输入数据的第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
结尾无空行

思路

陈越姥姥的PPT：

伪代码：

借鉴的别的博主的思路：链接
（1）题目增多了一个条件：当都是最短路径的时候选择便宜的

if(dist[v]+G[v][i]==dist[i]&&cost[v]+C[v][i]<cost[i]){
	//都是最短路径且花费小->更新花费
    cost[i]=cost[v]+C[v][i];
}

（2）找dist值最小的顶点v，这里使用了for循环比较，也可以用最小堆。值得注意的是，有可能找不到dist值最小的顶点，此时就要返回-1表示没有找到或者找完了，Dijsktra函数中就要对-1进行特判
（3）dist和cost的初始化，博主因为初始化成了-1，卡了很久！
初始化S的邻接点，如果能到i点，更新i点的dist为G[S][i]，更新cost为C[S][i];如果不能到这个点，就初始化为无穷大，因为要找最小路径，所以初始化为最大

for(int i=0;i<N;i++){
        if(G[S][i]){//S到i有边
            dist[i]=G[S][i];
            cost[i]=C[S][i];
        }else{//S到i无边（因为要找最小路径，所以初始化为最大）
            dist[i]=cost[i]=INIFITE;
        }
    }

Code

#include<iostream>
#include<queue>
#define MaxN 501
#define INIFITE 100001
int G[MaxN][MaxN];//记录两点之间的边
int C[MaxN][MaxN];//记录两点之间的花费
int dist[MaxN];//记录最短路径
int cost[MaxN];//记录价格
int visited[MaxN];
int N,M,S,D;
using namespace std;
//找当前dist中存储的最小值，返回它的v
int FindMin(){
    int index=-1,minlen=100001;
    for(int i=0;i<N;i++){
        if(!visited[i]&&dist[i]<minlen){
            minlen=dist[i];
            index=i;
        }
    }
    return index;
}
void Dijsktra(){
    visited[S]=true;
    dist[S]=0;//源头
    int v;
    while(1){
        v=FindMin();
        if(v==-1){//不存在最小v
            break;
        }
        visited[v]=true;
        for(int i=0;i<N;i++){
            if(!visited[i]&&G[v][i]){//没被访问过，而且是邻接点
                if(dist[v]+G[v][i]<dist[i]){
                    dist[i]=dist[v]+G[v][i];
                    cost[i]=cost[v]+C[v][i];
                }else if(dist[v]+G[v][i]==dist[i]&&cost[v]+C[v][i]<cost[i]){//都是最短路径且花费小->更新花费
                    cost[i]=cost[v]+C[v][i];
                }
            }
        }
    }
    printf("%d %d\n",dist[D],cost[D]);
}
int main(){
    scanf("%d %d %d %d",&N,&M,&S,&D);
    int v1,v2,len,price;
    for(int i=0;i<M;i++){
        scanf("%d %d %d %d",&v1,&v2,&len,&price);
        G[v1][v2]=G[v2][v1]=len;
        C[v1][v2]=C[v2][v1]=price;
    }
    for(int i=0;i<N;i++){
        visited[i]=false;
    }
    for(int i=0;i<N;i++){
        if(G[S][i]){//S到i有边
            dist[i]=G[S][i];
            cost[i]=C[S][i];
        }else{//S到i无边（因为要找最小路径，所以初始化为最大）
            dist[i]=cost[i]=INIFITE;
        }
    }
    Dijsktra();
    return 0;
}

其他类似问题

要求数最短路径时，将count数组初始化为1
若找到更短路->赋值：
count[W]=count[V];//前一个顶点有多少条路径，这个顶点就有多少条
若找到等长的路：
count[W]+=count[V];//又多一条最短路径count[w]=count[v]+count[w] (1)