WEEK 7 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

思路：

两次Dijkstra
题中有两种不同的线路，若没有商业线，我们可以直接考虑起点到终点的距离。由于商业线的存在，且只能使用一次，因此我们可以枚举每一条商业线(a,b)。我们从起点跑一次最短路，记录下dis1数组，然后从终点跑一次最短路，记录下dis2数组，然后对于当前商业线(a,b)，权值为w，其经过当前商业线的最短路径是min(dis1[a]+dis2[b]+w,dis1[b]+dis2[a]+w)。最后比较所有商业线与不使用商业线，取其最小值。
该题的重点在于如何输出，本人愚钝，所以在这里参考了大佬的代码：传送门

代码：

#include <iostream>
#include <queue>
#include <vector>
#include <string.h>

using namespace std;

const int inf = 1e8;
const int M = 1e5+10;
const int N = 505;
vector <int> v;
struct edge
{
	int to, next, w;
}es[M];
int head[N], dis1[N], dis2[N], path1[N], path2[N], vis[N];
int tot, n, m, k, s, e, ans, min1, min2, x, y, z, minpath = inf;
bool flag = false;

void add(int x,int y,int z)
{
	es[++tot].to = y;
	es[tot].next = head[x];
	head[x] = tot;
	es[tot].w = z;
}

priority_queue<pair<int, int> > q;

void dijkstra(int s,int *dis,int *path)
{
	memset(vis, 0, sizeof(vis));
	for (int i = 1; i <= n; i++)
		dis[i] = inf, path[i] = -1;
	while (q.size())
		q.pop();
	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 = es[i].next)
		{
			int y = es[i].to, w = es[i].w;
			if (dis[y] > dis[x] + w)
			{
				dis[y] = dis[x] + w;
				q.push(make_pair(-dis[y], y));
				path[y] = x;
			}
		}
	}
}

int main()
{
	while (scanf("%d %d %d", &n, &s, &e) != EOF)
	{
		ans = 0; tot = 0; minpath = inf;
		memset(head, 0, sizeof(head));
		scanf("%d", &m);
		for (int i = 1; i <= m; i++)
		{
			scanf("%d %d %d", &x, &y, &z);
			add(x, y, z);
			add(y, x, z);
		}
		dijkstra(s, dis1, path1);
		dijkstra(e, dis2, path2);
		scanf("%d", &k);
		for (int i = 1; i <= k; i++)
		{
			scanf("%d %d %d", &x, &y, &z);
			if (dis1[x] + dis2[y] + z < minpath) 
			{
				min1 = x; 
				min2 = y; 
				minpath = dis1[x] + dis2[y] + z; 
				ans = x;
			}
			if (dis2[x] + dis1[y] + z < minpath) 
			{
				min1 = y; 
				min2 = x; 
				minpath = dis2[x] + dis1[y] + z; 
				ans = y;
			}
		}
		if (!flag) 
			flag = true;
		else 
			putchar('\n');
		if(dis1[e]<minpath)
        {
            for (int i=e; i!=s; i=path1[i])
                v.push_back(i);
            v.push_back(s);
            for (int i=v.size()-1; i>=1; i--)
                printf("%d ",v[i]);
            printf("%d\nTicket Not Used\n%d\n",v[0],dis1[e]);
        }
        else 
		{
            for (int i=min1; i!=s; i=path1[i])
                v.push_back(i);
            v.push_back(s);
            for (int i=min2; i!=e; i=path2[i])
                v.insert(v.begin(),i);
            v.insert(v.begin(),e);
            for (int i=v.size()-1; i>=1; i--)
                printf("%d ",v[i]);
            printf("%d\n%d\n%d\n",v[0],ans,minpath);
        }
        v.clear();
	}
	return 0;
}