【从0开始的刷题生活】PTA数据结构2019-秋 07-图6 旅游规划 (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算法和Floyd算法，Floyd算法适用于多源最短路径的求解，所以使用Dijkstra算法最佳，由于还要记录价格信息，所以需要开两个数组 money[maxn] 和 dist[maxn] 来分别记录价格和路径长度

代码

#include <iostream>
#define maxn 600
#define infinity 30000
using namespace std;
typedef struct {
	int v1,v2;
	int length,money;
}Edge,*edge;//边的数据结构声明
typedef struct {
	int nv,ne;
	int arc[maxn][maxn];
	int cost[maxn][maxn];
}MGraph,*mgraph;//图的数据结构声明
int FindMinDist(MGraph g,int dist[],bool collected[])//初始化路径数组
{
	int minv,v;
	int mindist=infinity;
	for(v=0;v<g.nv;v++){
		if(collected[v]==false && dist[v]<mindist){
			mindist = dist[v];
			minv=v;
		}
	}
	if(mindist<infinity)
		return minv;
	else return -1;
}
bool Dijkstra(MGraph g,int dist[],int path[],int money[],int s)//s为起点,path[]数组记录了某一个点的前驱，dist[]和money[]分别记录路径长和价格总数
{
	bool collected[maxn];
	int v,w;
	for(v=0;v<g.nv;v++){ 
		dist[v]=g.arc[s][v];
		money[v]=g.cost[s][v];
		if(dist[v]<infinity)
			path[v]=s;
		else
			path[v]=-1;
		collected[v]=false;
	}
	dist[s]=0;
	money[s]=0;
	collected[s]=true;
	while(1){
		v=FindMinDist(g,dist,collected);
		if(v==-1) break;
		collected[v]=true;
		for(w=0;w<g.nv;w++)
			if(collected[w]==false && g.arc[v][w]<infinity){//如果某一个点没有被收录且某一个边还没有被收录
				if(g.arc[v][w]<0)
					return false;
				if(dist[v]+g.arc[v][w]<dist[w]){//到该点前一个点的路径加上其后面一条边的值比预估的该点的路经短
					dist[w]=dist[v]+g.arc[v][w];
					path[w]=v;
					money[w]=money[v]+g.cost[v][w];
				}
				else if((dist[v]+g.arc[v][w]==dist[w])&&(money[v]+g.cost[v][w]<money[w]))//实际的和预计的相等但价格比预计的低
				{
					money[w]=money[v]+g.cost[v][w];
				}
			}
	} 
	return true;
}	
int dist[maxn],path[maxn],money[maxn];	
int main()
{
	int n,m,s,d;
	int a,b,c,f;
	cin>>n>>m>>s>>d;
	MGraph g;
	g.nv=n;
	g.ne=m;
	for(int i=0;i<n;i++)
		for(int j=0;j<n;j++){ 
		g.arc[i][j]=infinity;
		g.cost[i][j]=infinity;
	} 
	for(int i=0;i<m;i++)
	{
		cin>>a>>b>>c>>f;
		g.arc[a][b]=c;
		g.arc[b][a]=c;
		g.cost[a][b]=f;
		g.cost[b][a]=f;
	}
	if(Dijkstra(g,dist,path,money,s))
	cout<<dist[d]<<" "<<money[d]<<endl;
}

写在后面

此题需要注意的是，能够看出用Dijkstra算法以及能够对Dijkstra算法进行适合的修改，颇有种"入乡随俗"的感觉