程序设计思维与实践 Week7 B-TT 的旅行日记

题目链接：B-TT的旅行日记

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

Input：
输入包含多组数据。每组数据第一行为 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 行是商业线路段的描述，格式同经济线。
所有路段都是双向的，但有可能必须使用商业车票才能到达机场。保证最优解唯一。

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

Sample Input：
4 1 4
4
1 2 2
1 3 3
2 4 4
3 4 5
1
2 4 3

Sample Output：
1 2 4
2
5

思路：
由于最多只有一站是商业线，那么可以枚举这条线路。其他的是经济线，找到这条路的两个端点a、b后，加上起点到a和b到终点的最短路就是所枚举的这种情况下的最优解，然后取最小值即可。经济线的最短路，就相当于单源最短路，对起点和终点都用dijkstral处理。

总结：
本题比较复杂，需要对路径的两端都进行处理，而且本题的输出要求十分严格，空格，换行的操作使得出现多次PE。

代码：

#include<bits/stdc++.h>
using namespace std;
const int INF=0x0fffffff;
int head[510],tot;
int dis[510];
int N,S,E;
int di[510],gr[510],did[510],grg[510];
struct edge{
	int to,next,w;
};
edge e[100000];
void init(){
	tot=0;
	memset(head,-1,sizeof(head));
}
void add(int x,int y,int w){
	e[++tot].to=y;
	e[tot].next=head[x];
	e[tot].w=w;
	head[x]=tot;
}
struct node{
	int num,v;
	bool operator<(const node& no)const{
		return v>no.v;
	}
};
priority_queue<node> q;
void dijkstra(int s,int* d,int* path){
	for(int i=1;i<=N;i++)
		d[i]=INF;
	while(!q.empty())
		q.pop();
	memset(dis,0,sizeof(dis));
	q.push((node){s,0});
	path[s]=-1;
	d[s]=0;
	while(!q.empty()){
		node current=q.top();
		q.pop();
		if(dis[current.num])
			continue;
		for(int i=head[current.num];i!=-1;i=e[i].next){
			int to_num=e[i].to;
			int val=e[i].w+current.v;
			if(val<d[to_num]){
				d[to_num]=val;
				path[to_num]=current.num;
				q.push((node){to_num,val});
			}
		}
	}
}
void print(int u,int* path){
	if(u==S){
		cout<<u;
		return;
	}
	print(path[u],path);
		cout<<" "<<u;
}
void print2(int u,int* path){
	if(u==E){
		cout<<" "<<u;
		return;
	}
	cout<<" "<<u;
	print2(path[u],path);
}
int main(){
	int M,K,xx,yy;
	bool flag=0;
	while(~scanf("%d%d%d",&N,&S,&E))
	{
	    if(flag)
	        cout<<endl;
		int stop=0;
		cin>>M;
		init();
		while(M--){
			int x,y,z;
			cin>>x>>y>>z;
			add(x,y,z);
			add(y,x,z);
		}
		dijkstra(S,di,did);
		dijkstra(E,gr,grg);
		int mint=di[E];
		cin>>K;
		while(K--){
			int x,y,z;
			cin>>x>>y>>z;
			int val=di[x]+gr[y]+z;
			if(val<mint){
				mint=val;
				stop=x;
				xx=x,yy=y;
			}
			swap(x,y);
			val=di[x]+gr[y]+z;
			if(val<mint){
				mint=val;
				stop=x;
				xx=x,yy=y;
			}
		}
		if(stop==0){
			print(E,did);
			cout<<endl<<"Ticket Not Used"<<endl;
		}
		else{
			print(xx,did);
			print2(yy,grg);;
			cout<<endl<<stop<<endl;
		}
		cout<<mint<<endl;
		flag=1;
	}
	return 0;
}