PTA——旅游规划

PTA——旅游规划

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

输入格式：

输入说明：输入数据的第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

分析：

这题要注意，不仅是要判断路径是否为最短，还要判断当路径的长度相等时，收费是否是最便宜的。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<map>
using namespace std;
const int inf=0x3f3f3f3f;
const int N=1001;
int pre[N],flag[N],a[N][N],dist[N][2];
map<pair<int,int>,int> city;
int n,m,s,d;
void dijkstra()
{
	pre[s]=-1;
	flag[s]=1;
	dist[s][0]=0;
	dist[s][1]=0;
	while(1){
		int min=inf,t;
		for(int i=0;i<n;++i){
			if(!flag[i]&&dist[i][0]<min){
				min=dist[i][0];
				t=i;
			}
		}
		if(t==d||min==inf){
			break;
		}
		flag[t]=1;
		for(int i=0;i<n;++i){
			if(!flag[i]&&dist[i][0]>dist[t][0]+a[t][i]){
				dist[i][0]=dist[t][0]+a[t][i];
				dist[i][1]=dist[t][1]+city[{t,i}];
				pre[i]=t;
			}
			else if(!flag[i]&&dist[i][0]==dist[t][0]+a[t][i]){
				if(dist[i][1]>dist[t][1]+city[{t,i}]){
					dist[i][1]=dist[t][1]+city[{t,i}];
					pre[i]=t;
				}
			}
		}
	}
}
int main()
{
	cin>>n>>m>>s>>d;
	for(int i=0;i<n;++i){
		for(int j=0;j<n;++j){
			a[i][j]=inf;
		}
	}
	for(int i=0;i<n;++i){
		dist[i][0]=inf;
		pre[i]=-1;
	}
	while(m--){
		int t1,t2,t3,t4;
		cin>>t1>>t2>>t3>>t4;
		a[t1][t2]=a[t2][t1]=t3;
		city[{t1,t2}]=city[{t2,t1}]=t4;
	}
	for(int i=0;i<n;++i){
		if(a[s][i]<inf){
			dist[i][0]=a[s][i];
			dist[i][1]=city[{s,i}];
			pre[i]=s;
		}
	}
	dijkstra();
	printf("%d %d\n",dist[d][0],dist[d][1]);
	return 0;
}