程序设计思维与实践 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
解题思路：
题目给定了起点与终点，而且要求商业线最多乘坐依次。
可以枚举每一条商业线，计算起点origin到商业线起点bg的最短路以及ed到终点terminal的最短路再加上该商业线所花费的时间。
以origin为起点，运用dijkstra算法求出单源最短路，得出d1数组
以terminal为源点，求出单源最短路，得出d2数组。
当走商业线时，取min{d1[u]+d2[v]+w,d1[v]+d2[u]+w}.再与不走商业线取最小值。
除此之外，这道题还需要依次输出走过的每一个车站。last1数组从origin出发。last2数组从terminal出发。数组记录的是当前点的前序点。所以可以用terminal出发的last2数组来输出。或者用last1数组倒过来输出。

if (ans == d1[terminal]) {    //用last2数组输出
			int now = origin;
			while (now != terminal) {
				printf("%d ", now);
				now = e[last2[now]].from;
			}
			printf("%d\n", terminal);
			printf("Ticket Not Used\n");
			printf("%d\n",ans);
		}

while (now != origin) {  //找到origin到bg的路径
				go[++cnt] = now;
				now = e[last1[now]].from;
			}
			go[++cnt] = origin;
			for (int i = cnt; i >= 1; i--) printf("%d ", go[i]);

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <vector>
using namespace std;
const int MAXN = 505, INF=0x3f3f3f3f;
int n, origin, terminal, m, k;

struct edge {
	int from, to, dist;
	edge (int from, int to, int dist) : from(from), to(to), dist(dist) {}
};
struct node {
	int u, d;
	node(int u, int d) : u(u), d(d) {}
	bool operator<(const node &minheap) const
	{
		return d >minheap.d;
	}
};
vector<edge> e;
vector<int> g[MAXN];
int d1[MAXN], d2[MAXN], last1[MAXN], last2[MAXN];
bool done[MAXN];
priority_queue<node> q;
int com[MAXN][MAXN];
void dijkstra(int x)
{
	memset(done, false, sizeof(done));
	if (x == origin){
		memset(d1, 0x3f, sizeof(d1));
		d1[x] = 0;
	}
	else {
		memset(d2, 0x3f, sizeof(d2));
		d2[x] = 0;
	}
	q.push(node(x, 0));
	while (!q.empty()) {
		int u = q.top().u; q.pop();
		if (done[u]) continue;
		done[u] = true;
		for (int i = 0; i < g[u].size(); i++) {
			edge &now = e[g[u][i]];
			if (x == origin && d1[now.to] > d1[u] + now.dist) {
				d1[now.to] = d1[u] + now.dist;
				last1[now.to] = g[u][i];	
				q.push(node(now.to, d1[now.to]));
			}
			else
			if (x == terminal && d2[now.to] > d2[u] + now.dist) {
				d2[now.to] = d2[u] + now.dist;
				last2[now.to] = g[u][i];
				q.push(node(now.to, d2[now.to]));
			}
		}
	}
}
int main()
{
	int flag =0;
	while (scanf("%d %d %d", &n, &origin, &terminal)!=EOF) {
		if (flag++)  
			printf("\n");
		e.clear();
	for (int i = 1; i <= n; i++) g[i].clear();

		scanf("%d", &m);
		int a, b, c;
		for (int i = 1; i <= m; i++) {
			scanf("%d %d %d", &a, &b, &c);
		e.push_back(edge(a, b, c));
	    g[a].push_back(e.size() - 1);
		e.push_back(edge(b, a, c));
	    g[b].push_back(e.size() - 1);
		}	
		memset(com, 0x3f, sizeof(com));
		scanf("%d", &k);
		for (int i = 1; i <= k; i++) {
			scanf("%d %d %d", &a, &b, &c);
			com[a][b] = min(com[a][b], c);
			com[b][a] = com[a][b];
		}
		dijkstra(origin);
		dijkstra(terminal);
		int bg, ed, ans = d1[terminal];
		for (int i = 1; i <= n; i++)
			for (int j = i + 1; j <= n; j++) {
				if (com[i][j] == INF) continue;
				if (ans > d1[i] + d2[j] + com[i][j]) {
					ans = d1[i] + d2[j] + com[i][j];
					bg = i;
					ed = j;
				}
				if (ans > d1[j] + d2[i] + com[i][j]) {
					ans = d1[j] + d2[i] + com[i][j];
					bg = j;
					ed = i;
				}
			}
		if (ans == d1[terminal]) {
			int now = origin;
			while (now != terminal) {
				printf("%d ", now);
				now = e[last2[now]].from;
			}
			printf("%d\n", terminal);
			printf("Ticket Not Used\n");
			printf("%d\n",ans);
		}
		else {
			int cnt = 0, now = bg;
	    	int go[MAXN];
			while (now != origin) {
				go[++cnt] = now;
				now = e[last1[now]].from;
			}
			go[++cnt] = origin;
			for (int i = cnt; i >= 1; i--) printf("%d ", go[i]);
			now = ed;
			while (now != terminal) {
				printf("%d ", now);
				now = e[last2[now]].from;
			}
			printf("%d\n%d\n%d\n", terminal, bg, ans);
		}
	}
	return 0;
}