【Week7 作业B】TT的旅行日记

题目描述

众所周知，TT 有一只魔法猫。
今天他在 B 站上开启了一次旅行直播，记录他与魔法猫在喵星旅游时的奇遇。 TT 从家里出发，准备乘坐猫猫快线前往喵星机场。猫猫快线分为经济线和商业线两种，它们的速度与价钱都不同。当然啦，商业线要比经济线贵，TT 平常只能坐经济线，但是今天 TT 的魔法猫变出了一张商业线车票，可以坐一站商业线。假设 TT 换乘的时间忽略不计，请你帮 TT 找到一条去喵星机场最快的线路，不然就要误机了！

输入格式

输入包含多组数据。每组数据第一行为 3 个整数 N, S 和 E (2 ≤ N ≤ 500, 1 ≤ S, E ≤ 100)，即猫猫快线中的车站总数，起点和终点（即喵星机场所在站）编号。
下一行包含一个整数 M (1 ≤ M ≤ 1000)，即经济线的路段条数。
接下来有 M 行，每行 3 个整数 X, Y, Z (1 ≤ X, Y ≤ N, 1 ≤ Z ≤ 100)，表示 TT 可以乘坐经济线在车站 X 和车站 Y 之间往返，其中单程需要 Z 分钟。
下一行为商业线的路段条数 K (1 ≤ K ≤ 1000)。
接下来 K 行是商业线路段的描述，格式同经济线。
所有路段都是双向的，但有可能必须使用商业车票才能到达机场。保证最优解唯一。

输入样例

4 1 4
4
1 2 2
1 3 3
2 4 4
3 4 5
1
2 4 3

输出格式

对于每组数据，输出3行。第一行按访问顺序给出 TT 经过的各个车站（包括起点和终点），第二行是 TT 换乘商业线的车站编号（如果没有使用商业线车票，输出"Ticket Not Used"，不含引号），第三行是 TT 前往喵星机场花费的总时间。
本题不忽略多余的空格和制表符，且每一组答案间要输出一个换行

输出样例

1 2 4
2
5

思路

题中有两种不同的线路，若没有商业线，我们可以直接考虑起点到终点的距离。由于商业线的存在，且只能使用一次，所以我们可以枚举每一条商业线（u,v），其耗时为w，假如我们知道起点到u,v的距离dis1[u],dis1[v]以及终点到u,v的距离dis2[u],dis2[v]，取min(dis1[u]+dis2[v]+w,dis1[v]+dis2[u]+w)，枚举完所有商业线后，取得耗时最小的线路，再与不使用商业线的耗时比较，即可取到最小耗时的线路。接下来的问题就是如何求起点到各点，以及终点到各点的最短路。
这里采用Dijkstra算法

Dijkstra

用途

解决图中无负边的单源最短路问题。

流程

从源端s出发，dis[a]表示s到a的最短距离，初始化所有dis[a]为无穷大，将s压入最小堆中，每次从堆中取出距源点最近的点，遍历该点所连接的边，更新该点的邻接点y到源点的距离为min(dis[y],dis[x]+w)(w为边xy的距离)，直到最小堆为空，此时dis数组中即为各点距源点的最短距离。在求取距离的过程中，可通过记录各点到源点路径的上一点已达到记录路径的目的。

代码

#include <iostream>
#include<queue>
#include<string.h> 
using namespace std;
const int size=5e3+10;
const int N=500+5;
const int inf=1e9;
struct Edge{
	int to,next,w;
}e[size]; 
Edge other[size];
int head[N],tot,n,vis[N],spre[N],sdis[N],tpre[N],tdis[N];
void add(int x,int y,int w)
{
	e[++tot].to=y,e[tot].next=head[x];
	e[tot].w=w;head[x]=tot;
}
priority_queue<pair<int,int>>q;
void dijkstra(int s,int*dis,int*pre)
{
	while(!q.empty())q.pop();
	memset(vis,0,sizeof(vis));
	for(int i=1;i<=n;i++)
		dis[i]=inf,pre[i]=-1;
	dis[s]=0;pre[s]=s;
	q.push(make_pair(0,s));
	while(!q.empty())
	{
		int x=q.top().second;
		q.pop();
		if(vis[x])continue;
		vis[x]=1;
		for(int i=head[x];i!=-1;i=e[i].next)
		{
			int y=e[i].to,w=e[i].w;
			if(dis[y]>dis[x]+w)
			{
				dis[y]=dis[x]+w;
				pre[y]=x;
				q.push(make_pair(-dis[y],y));
			}
		}
	}
}
void print(int to,int*pre)
{
	if(pre[to]!=to)
	{
		print(pre[to],pre);
		printf(" %d",to);		
	}
	else
	{
		printf("%d",to);
	}
}
int main(int argc, char** argv) {
	int s,e;
	int number=1;
	while(~scanf("%d%d%d",&n,&s,&e))
	{
		if(number!=1)
			printf("\n");
		number++;
		memset(head,-1,sizeof(head));
		tot=-1;
		int m;
		scanf("%d",&m);
		while(m--)
		{
			int x,y,z;
			scanf("%d%d%d",&x,&y,&z);
			add(x,y,z);
			add(y,x,z);
		}
		int k;
		scanf("%d",&k);
		for(int i=0;i<k;i++)
		{
			int x,y,z;
			scanf("%d%d%d",&x,&y,&z);
			other[i].next=x,other[i].to=y,other[i].w=z;
		}
		dijkstra(s,sdis,spre);
		dijkstra(e,tdis,tpre);
		int point=-1,distance=sdis[e];
		bool sort=true;
		for(int i=0;i<k;i++)
		{
			if(sdis[other[i].next]+tdis[other[i].to]+other[i].w<distance)
			{
				distance=sdis[other[i].next]+tdis[other[i].to]+other[i].w;
				point=i;sort=true;
			}
			if(sdis[other[i].to]+tdis[other[i].next]+other[i].w<distance)
			{
				distance=sdis[other[i].to]+tdis[other[i].next]+other[i].w;
				point=i;sort=false;
			}
		}
		if(point==-1)
		{
			print(e,spre);
		}
		else
		{
			if(sort)
			{
				print(other[point].next,spre);
				int temp=other[point].to;
				while(tpre[temp]!=temp)
				{
					printf(" %d",temp);
					temp=tpre[temp];
				}
				printf(" %d",temp);
			}
			else
			{
				print(other[point].to,spre);
				int temp=other[point].next;
				while(tpre[temp]!=temp)
				{
					printf(" %d",temp);
					temp=tpre[temp];
				}
				printf(" %d",temp);
			}
		}
		printf("\n");
		if(point==-1)
			printf("Ticket Not Used\n");
		else
		{
			if(sort)
				printf("%d\n",other[point].next);
			else
				printf("%d\n",other[point].to);
		}
		printf("%d\n",distance);
	}

	return 0;
}