PTA天梯地图

本题要求你实现一个天梯赛专属在线地图,队员输入自己学校所在地和赛场地点后,该地图应该推荐两条路线:一条是最快到达路线;一条是最短距离的路线。题目保证对任意的查询请求,地图上都至少存在一条可达路线。

输入格式:

输入在第一行给出两个正整数N(2 \le N \le 500)和M,分别为地图中所有标记地点的个数和连接地点的道路条数。随后M行,每行按如下格式给出一条道路的信息:

V1 V2 one-way length time

其中V1V2是道路的两个端点的编号(从0到N-1);如果该道路是从V1V2的单行线,则one-way为1,否则为0;length是道路的长度;time是通过该路所需要的时间。最后给出一对起点和终点的编号。

输出格式:

首先按下列格式输出最快到达的时间T和用节点编号表示的路线:

Time = T: 起点 => 节点1 => ... => 终点

然后在下一行按下列格式输出最短距离D和用节点编号表示的路线:

Distance = D: 起点 => 节点1 => ... => 终点

如果最快到达路线不唯一,则输出几条最快路线中最短的那条,题目保证这条路线是唯一的。而如果最短距离的路线不唯一,则输出途径节点数最少的那条,题目保证这条路线是唯一的。

如果这两条路线是完全一样的,则按下列格式输出:

Time = T; Distance = D: 起点 => 节点1 => ... => 终点

输入样例1:

10 15
0 1 0 1 1
8 0 0 1 1
4 8 1 1 1
5 4 0 2 3
5 9 1 1 4
0 6 0 1 1
7 3 1 1 2
8 3 1 1 2
2 5 0 2 2
2 1 1 1 1
1 5 0 1 3
1 4 0 1 1
9 7 1 1 3
3 1 0 2 5
6 3 1 2 1
5 3

输出样例1:

Time = 6: 5 => 4 => 8 => 3
Distance = 3: 5 => 1 => 3

输入样例2:

7 9
0 4 1 1 1
1 6 1 3 1
2 6 1 1 1
2 5 1 2 2
3 0 0 1 1
3 1 1 3 1
3 2 1 2 1
4 5 0 2 2
6 5 1 2 1
3 5

输出样例2:

Time = 3; Distance = 4: 3 => 2 => 5

题意讲的很清楚了没什么好说的了,看代码

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 1e4+10;
/*----------------------------------------------------------------head------------------------------------------------------------------*/
int n,m,ss,dd;
int T[510][510],D[510][510],dist[510],disd[510],vist[510],visd[510],pret[510]/*记录最小时间的前驱*/,pred[510]/*记录最短距离的前驱*/,SUM[510]/*记录节点个数*/;
int u,v,is,d,t;
void print(int cnt,int c[]){//打印路径
	for(int i = cnt-1; i >= 0; i--){
		if(i != cnt-1)
			printf(" =>");
		printf(" %d",c[i]);
	}
	puts("");
}
/*------------------------------------------------------------dijkstra-----------------------------------------------------------------*/
//求解最小时间的最短路
void dijkstraT(){
	for(int i = 0; i < n; i++){//初始化
		dist[i] = T[ss][i];
		disd[i] = D[ss][i];
		vist[i] = visd[i] = 0;
		pret[i] = pred[i] = ss;
	}
	dist[ss] = disd[ss] = 0;//初始化
	vist[ss] = visd[ss] = 1;//初始化
	pret[ss] = -1;//初始化
	int Min,pos;
	for(int i = 1; i < n; i++){
		Min = INF;
		for(int j = 0; j < n; j++){
			if(!vist[j] && dist[j]dist[pos]+T[pos][j]){
				dist[j] = dist[pos]+T[pos][j];
				disd[j] = disd[pos]+D[pos][j];
				pret[j] = pos;
			}
			else if(!vist[j] && dist[j] == dist[pos]+T[pos][j]){
				if(disd[j] > disd[pos]+D[pos][j]){//时间相同选距离最短
					disd[j] = disd[pos]+D[pos][j];
					pret[j] = pos;
				}
			}
		}
	}
}
//求解最短距离的最短路
void dijkstraD(){
	for(int i = 0; i < n; i++){//初始化
		disd[i] = D[ss][i];
		visd[i] = 0;
		pred[i] = ss;
		SUM[i] = 1;
	}
	disd[ss] = 0;//初始化
	visd[ss] = 1;//初始化
	pred[ss] = -1;//初始化
	int Min,pos;
	for(int i = 1; i < n; i++){
		Min = INF;
		for(int j = 0; j < n; j++){
			if(!visd[j] && disd[j]disd[pos]+D[pos][j]){
				disd[j] = disd[pos]+D[pos][j];
				pred[j] = pos;
				SUM[j] = SUM[pos]+1;
			}
			else if(!visd[j] && disd[j]==disd[pos]+D[pos][j]){//距离相同选节点数最小
				if(SUM[j] > SUM[pos]+1){
					SUM[j] = SUM[pos]+1;
					pred[j] = pos;
				}
			}
		}
	}
}
/*------------------------------------------------------------main--------------------------------------------------------------------*/
int main(){
	scanf("%d %d",&n,&m);
	memset(T,INF,sizeof(T));
	memset(D,INF,sizeof(D));
	while(m--){
		scanf("%d %d %d %d %d",&u,&v,&is,&d,&t);
		if(is){
			if(T[u][v]>t)//去重
				T[u][v] = t;
			if(D[u][v]>d)//去重
				D[u][v] = d;
		}
		else{
			if(T[u][v]>t)//去重
				T[u][v] = T[v][u] = t;
			if(D[u][v]>d)//去重
				D[u][v] = D[v][u] = d;
		}
	}
	scanf("%d %d",&ss,&dd);
	dijkstraT();
	dijkstraD();
	int pt = dd;
	int pd = dd;
	int flag = 1;//flag==1表示两条路径相同,不然不同。
	int a[510],b[510],cnta,cntb;
	cnta = cntb = 0;
	while(pt != -1){
		a[cnta++] = pt;
		pt = pret[pt];
	}
	while(pd != -1){
		b[cntb++] = pd;
		pd = pred[pd];
	}
	if(cnta != cntb)//节点数不同路径肯定不同
		flag = 0;
	else{
		for(int i = 0; i < cnta; i++){
			if(a[i] != b[i]){
				flag = 0;
				break;
			}
		}
	}
	if(flag){
		printf("Time = %d; Distance = %d:",dist[dd],disd[dd]);
		print(cnta,a);
	}
	else{
		printf("Time = %d:",dist[dd]);
		print(cnta,a);
		printf("Distance = %d:",disd[dd]);
		print(cntb,b);
	}
  	return 0;
}


你可能感兴趣的:(PTA)