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

思路

解法

1、从甲地开始做一遍迪杰斯特拉，记录下在不乘坐商业线的情况下各站点到甲地的最短路径以及前驱节点，从乙地开始做一遍同样的操作，然后枚举各商业站(u,v,w)，获得min{dis1[u]+dis2[v]+w, dis1[v]+dis2[u]+w，dis1[e]}。在输出从起点到终点经过的站点时，需要用遍历记录前驱节点的数组并用一个新的数组存放遍历得到的元素，输出时要注意前后顺序。
2、因为priority_queue<pair<int,int> >为最大堆，若想使其变为最小堆，可以使用 q.push(make_pair(-dis[y],y));
3、pair<int,int>默认按照first来排序。

错误

1、注意判断边池应开的大小，虽然输入为n条边，但考虑到对每条边都需要在边池中增加两条边，故需要将边池开为2倍的增加边数。
2、注意到v1、v2可能就是s或e所以不可直接用pre[v1/v2]==s，e来作为while的结束条件，考虑可以将v1、v2的前驱节点赋为-1，然后利用pre[v1/v2]= =-1来作为while的终止条件
3、注意此题严格控制输出格式，需要在每组数据之间输出一个换行，而在所有数据结束处不输出换行符。
4、要使用ios::sync_with_stdio(false);关同步，不然会TE。

代码

#include<iostream>
#include<bits/stdc++.h>
using namespace std;
const int maxm=2010;//注意虽然题目给出输入1000条无向边，但因为要往边池中加入两个方向的边，所以应该将边池开为两倍大小 
const int maxn=501;
const int inf=5*1e8;
struct Edge
{
	int to,next,w;
}e[maxm];
int head[maxn],tot,n,m,dis1[maxn],dis2[maxn],pre1[maxn],pre2[maxn];
bool vis[maxn]; 
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.size())
		q.pop();
	memset(vis,0,sizeof(vis));
	for(int i=1;i<=n;i++)
		dis[i]=inf;
	pre[s]=-1;
	dis[s]=0;
	q.push(make_pair(0,s));
	while(q.size())
	{
		int x=q.top().second;
		q.pop();
		if(vis[x]) continue;
		vis[x]=1;
		for(int i=head[x];i;i=e[i].next)
		{
			int y=e[i].to,w=e[i].w;
			if(dis[y]>dis[x]+w)
			{
				pre[y]=x;
				dis[y]=dis[x]+w;
				q.push(make_pair(-dis[y],y));
			}
		}
	}
}
int main()
{
	int s,e,k,first=0;
	ios::sync_with_stdio(false);
	while((scanf("%d%d%d",&n,&s,&e))!=EOF)
	{
		if(first) 		
			printf("\n");//不要忘记每组数据之间还要输出一个空行。 
		first=1;
		scanf("%d",&m);
		tot=1;
		memset(head,0,sizeof(head));
		for(int i=0;i<m;i++)
		{
			int x,y,z;
			scanf("%d%d%d",&x,&y,&z);
			add(x,y,z);
			add(y,x,z);
		}
		dijkstra(s,dis1,pre1);
		dijkstra(e,dis2,pre2);
		int len=dis1[e];
		int v1,v2,flag=0;
		scanf("%d",&k);
		for(int i=0;i<k;i++)
		{
			int x,y,z;
			scanf("%d%d%d",&x,&y,&z);
			int len1=dis1[x]+z+dis2[y];
			if(len1<len)
			{
				len=len1;
				flag=1;
				v1=x;
				v2=y;
			}
			len1=dis1[y]+z+dis2[x];
			if(len1<len)
			{
				len=len1;
				flag=1;
				v1=y;
				v2=x;
			}
		}
		if(flag==0)
		{
			int temp=s;
			while(pre2[temp]!=-1)
			{
			printf("%d ",temp);
			temp=pre2[temp];				
			}
			printf("%d\n",temp);
			printf("Ticket Not Used\n");
			printf("%d\n",len);			
		}
		else
		{
			int temp[maxn];
			int i=0;
			temp[i]=v1;

			while(pre1[temp[i]]!=-1)
			{
				i++;
				temp[i]=pre1[temp[i-1]];
			}				
			for(i;i>=0;i--)
				printf("%d ",temp[i]);
			int temp1;
			temp1=v2;
			while(pre2[temp1]!=-1)
			{
				printf("%d ",temp1);
				temp1=pre2[temp1];
			}
			printf("%d\n",temp1);				
			//第二行 
			printf("%d\n",v1);
			//第三行 
			printf("%d\n",len);
		}

	}
	
	return 0;
}