PTA 7-25 旅游规划 Dijkstra

PTA 7-25 旅游规划 Dijkstra

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

输入格式:

输入说明：输入数据的第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 扩展一下即可，我使用了 邻接矩阵 和 邻接表（vector实现）两种数据结构储存图。

邻接矩阵：

#include <stdio.h>
#define INF 99999999
struct road{
	int lenth;
	int money;
};
int main()
{
	struct road map[500][500];
	int dis[500],book[500],pay[500];
	int n,m,s,d,i,j,min,index;
	scanf ("%d%d%d%d",&n,&m,&s,&d);
	for (i=0;i<n;i++)//初始化 
	{
		for (j=0;j<n;j++)
		{
			if (i==j)
				map[i][j].lenth=0;
			else
				map[i][j].lenth=INF;
			map[i][j].money=0;
		}
		book[i]=0; 
	}
	int index1,index2,lenth,money;
	for (i=0;i<m;i++)//输入权值 
	{
		scanf ("%d%d%d%d",&index1,&index2,&lenth,&money);
		map[index1][index2].lenth=lenth;
		map[index2][index1].lenth=lenth;
		map[index1][index2].money=money;
		map[index2][index1].money=money;
	}
	for (i=0;i<n;i++)//将初始值放入dis[]和money[]中 
	{
		dis[i]=map[s][i].lenth;
		pay[i]=map[s][i].money;
	}
	book[s]=1;//标记初始点为确定值 
	for (i=0;i<n;i++)//寻找估计值中的最小值 
	{
		min=INF;
		for (j=0;j<n;j++)
		{
			if (book[j]==0&&dis[j]<min)
			{
				min=dis[j];
				index=j;
			}
		}
		book[index]=1;//标记为确定值 
		for (j=0;j<n;j++)
		{
			if (map[index][j].lenth!=INF)//松弛估计值 
			{
				if (dis[j]>dis[index]+map[index][j].lenth)
				{
					dis[j]=dis[index]+map[index][j].lenth;
					pay[j]=pay[index]+map[index][j].money;
				}
				if (dis[j]==dis[index]+map[index][j].lenth&&pay[j]>pay[index]+map[index][j].money)
					pay[j]=pay[index]+map[index][j].money;
			}
		}
	}
	printf ("%d %d",dis[d],pay[d]);
	return 0;
}

邻接表：

#include <iostream>
#include <vector>
#define INF 99999999
using namespace std;
struct Node{
	int to;//下一节点的编号 
	int lenth;//公路长度 
	int pay;//花费 
};
int main()
{
	vector <struct Node> map[500];
	int dis[500],book[500]={0},money[500]={0};//dis[i]为到达i节点的最短路径，money[i]为到达该节点的花费 
	int n,m,s,d,i,j,min,index,index1,index2,lenth,pay;
	cin>>n>>m>>s>>d;
	for (i=0;i<m;i++)//输入边和边的权 
	{
		struct Node temp;
		cin>>index1>>index2>>lenth>>pay;
		temp.to=index2;
		temp.lenth=lenth;
		temp.pay=pay;
		map[index1].push_back(temp);
		temp.to=index1;//无向图 
		map[index2].push_back(temp);
	}
	for (i=0;i<n;i++)//初始化dis[]数组 
		dis[i]=INF;
	dis[s]=0;
	for (i=0;i<map[s].size();i++)//将初始值放入dis[]和money[]中 
	{
		dis[map[s][i].to]=map[s][i].lenth;
		money[map[s][i].to]=map[s][i].pay;
	}
	book[s]=1;//标记初始点为确定值 
	for (i=0;i<n;i++)//寻找估计值中最小值 
	{
		min=INF;
		index=-1;
		for (j=0;j<n;j++)
		{
			if (book[j]==0&&dis[j]<min)
			{
				min=dis[j];
				index=j;
			}
		}
		if (index==-1)
			break;
		book[index]=1;//标记为确定值 
		for (j=0;j<map[index].size();j++)//松弛估计值 
		{
			if (dis[map[index][j].to]>dis[index]+map[index][j].lenth)
			{
				dis[map[index][j].to]=dis[index]+map[index][j].lenth;
				money[map[index][j].to]=money[index]+map[index][j].pay;
			}
			if (dis[map[index][j].to]==dis[index]+map[index][j].lenth&&money[map[index][j].to]>money[index]+map[index][j].pay)
				money[map[index][j].to]=money[index]+map[index][j].pay;
		}
	}
	cout<<dis[d]<<" "<<money[d];
	return 0;
}