【Dijkstra】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 行是商业线路段的描述，格式同经济线。
所有路段都是双向的，但有可能必须使用商业车票才能到达机场。保证最优解唯一。

输出

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

样例

输入样例

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

输出样例

1 2 4
2
5

思路

由于只让乘坐一条商业线，且商业线数量不多（<=1000）,所以我们在可以采用枚举的方式，先用dijkstra求出从原点到其他车站的最短距离ds[]，再求出从终点到其他车站的最短距离de[]，没有商业线的最短距离就是ans=ds[e]，之后枚举商业线，则最短路线ans=min(ans,ds[x]+ds[y]+w,ds[y]+ds[x]+w)。
由于还需要输出路径，我们需要在dij的过程中记录前驱pre，最终输出时，若不乘坐商业线，直接递归遍历e到s，否则分两步输出，首先递归i到s，再递推j到e（i代表商业线的起始站点，j代表商业线的到达站点）。

代码

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<iomanip>
#include<queue> 
using namespace std;
#define rep(i,s,t) for(int i=s;i<=t;i++)
#define mem(a,s) memset(a,s,sizeof(a))
const int inf=0x3f3f3f3f;//!
const int M=2e3+10;//两倍
const int N=5e2+5;
struct edge{
	int v,next,w;
}e[M];
int head[N],tot,S,E,n,m,dis[N][2],pre[N][2];//?两个
bool vis[N][2];
void add_edge(int x,int y,int z){e[++tot].v=y;e[tot].w=z;e[tot].next=head[x],head[x]=tot;}
priority_queue<pair<int,int> >q;//负数最小堆
void dijkstra(int s,int x){
	while(!q.empty()) q.pop();
	for(int i=1;i<=n;i++)
			vis[i][x]=0,dis[i][x]=inf,pre[i][x]=0;
	dis[s][x]=0;
	q.push({0,s});
	while(!q.empty()){
		int u=q.top().second;
		q.pop();
		if(vis[u][x])continue;
		vis[u][x]=1;
		for(int i=head[u];i;i=e[i].next){
			int v=e[i].v,w=e[i].w;
			if(dis[v][x]>dis[u][x]+w){
				pre[v][x]=u;
				dis[v][x]=dis[u][x]+w;
				q.push({-dis[v][x],v});
			}
		}
	}
}
void print1(int x){
	if(x==S)
		cout<<S;
	else{
		print1(pre[x][0]);
		cout<<' '<<x;
	}
}
void print2(int x){
	if(x==E)
		cout<<' '<<E;
	else{
		cout<<' '<<x;
		print2(pre[x][1]);
	}
}
int main(){
//	freopen("1.txt","r",stdin);
	int x,y,z,K;
	bool cnt=false;
	while(~scanf("%d%d%d%d",&n,&S,&E,&m)){
		if(cnt)
			cout<<endl;
		else cnt=true;
		tot=0;
		rep(i,1,n) head[i]=0;
		rep(i,1,m){
			scanf("%d%d%d",&x,&y,&z);
			add_edge(x,y,z);add_edge(y,x,z);
		}
		dijkstra(S,0);
		dijkstra(E,1);
		int ans=dis[E][0],ansi=E,ansj=E;
		scanf("%d",&K);
		rep(i,1,K){
			scanf("%d%d%d",&x,&y,&z);
			if(ans>dis[x][0]+dis[y][1]+z)
				ans=dis[x][0]+dis[y][1]+z,ansi=x,ansj=y;
			if(ans>dis[y][0]+dis[x][1]+z)
				ans=dis[y][0]+dis[x][1]+z,ansi=y,ansj=x;			
		}
		if(ans==dis[E][0])
			print1(E);
		else{
			print1(ansi);
			print2(ansj);
		}
		cout<<endl;
		if(ans==dis[E][0]) cout<<"Ticket Not Used"<<endl;
		else cout<<ansi<<endl;
		cout<<ans<<endl;
	}
	

	return 0;
}