程序设计思维 B - TT 的旅行日记（Dijkstra算法）

题目

众所周知，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算法
Dijkstra算法是一种贪心算法，用于解决图中没有负边的单源最短路问题 （求图中给定一点到其他点的最短路径）。假设给定源点s，使用Dijkstra算法要用到三个一维数组和一个优先队列（最小堆）：
dis[i]：从点s到点i的最短距离
pre[i] = k：从点s到点i之前要经过点k
visit[i]：是否有访问过点i
优先队列q：每次从中取出距离点s最短的点
初始时，dis数组都置为 +∞，visit数组都置为false，pre数组都置为-1。
Dijkstra算法的主体：

void Dijkstra(int s, int* dis, int* pre)
{
	dis[s] = 0;
	q.push(make_pair(0, s));

	while (!q.empty()) {
		int x = q.top().second;
		q.pop();

		if (visit[x])continue;

		visit[x] = true;

		for (int i = Head[x]; i != -1; i = Edges[i].next) {
			int y = Edges[i].to;
			int w = Edges[i].weight;

			if (dis[y] > dis[x] + w) {
				dis[y] = dis[x] + w;
				pre[y] = x;
				q.push(make_pair(-dis[y], y));

			}
		}
	}
}

二、解题
（1）以起点s为源点求单源最短路，将结果存放在数组dis1中。
（2）以终点e为源点求单源最短路，将结果存放在数组dis2中。
（3）对于每一条商业线(u, v, w)，取 Min = min(dis1[u]+dis2[v]+w, dis1[v]+dis2[u]+w)，同时更新换成商业线的起点和终点。
（4）如果没有换乘，则输出dis1的结果，否则输出换乘的结果。

代码

#include<iostream>
#include <utility>
#include<queue>
#include<cstring>
#include<iostream>
#define MAX_V 1000000
#define MAX_E 1000000
#define Inf 100000000
using namespace std;

int N, S, E;
int M, K;

int Min;
int changeStart, changeEnd;

struct edge {
	int to;
	int weight;
	int next;
}Edges[MAX_E];
int Head[MAX_V];
int tot = 0;

priority_queue<pair<int, int> > q;
int dis1[MAX_V];
int dis2[MAX_V];

int pre1[MAX_V];
int pre2[MAX_V];

bool visit[MAX_V];

int Array[MAX_V];

void Init0()
{
	while (!q.empty()) q.pop();

	for (int i = 0; i < MAX_V; i++) Head[i] = -1;
	tot = 0;

	changeStart = 0;
	changeEnd = 0;
}

void Init1()
{
	while (!q.empty()) q.pop();
	for (int i = 0; i < MAX_V; i++) {
		visit[i] = false;
		dis1[i] = Inf;
		pre1[i] = -1;
	}
}

void Init2()
{
	while (!q.empty()) q.pop();
	for (int i = 0; i < MAX_V; i++) {
		visit[i] = false;
		dis2[i] = Inf;
		pre2[i] = -1;
	}
}

void add(int u, int v, int w)
{
	tot++;
	Edges[tot].to = v;
	Edges[tot].weight = w;
	Edges[tot].next = Head[u];
	Head[u] = tot;
}

void Dijkstra(int s, int* dis, int* pre)
{
	dis[s] = 0;
	q.push(make_pair(0, s));

	while (!q.empty()) {
		int x = q.top().second;
		q.pop();

		if (visit[x])continue;

		visit[x] = true;

		for (int i = Head[x]; i != -1; i = Edges[i].next) {
			int y = Edges[i].to;
			int w = Edges[i].weight;

			if (dis[y] > dis[x] + w) {
				dis[y] = dis[x] + w;
				pre[y] = x;
				q.push(make_pair(-dis[y], y));

			}
		}
	}
}

int main()
{
	int round = 0;

	// cin >> N >> S >> E
	// scanf("%d%d%d", &N, &S, &E) != EOF
	while (scanf("%d %d %d", &N, &S, &E) != EOF)
	{
		if (round) cout << endl;
		round++;

		Init0();

		//scanf("%d", &M);
		cin >> M;

		for (int i = 0; i < M; i++) {
			int u, v, w;
			// cin >> u >> v >> w;
			scanf("%d %d %d", &u, &v, &w);
			add(u, v, w);
			add(v, u, w);
		}

		Init1();
		Dijkstra(S, dis1, pre1);

		Init2();
		Dijkstra(E, dis2, pre2);

		scanf("%d", &K);
		// cin >> K;

		int Min = dis1[E];

		int changeStart = 0;
		int changeEnd = 0;

		for (int i = 0; i < K; i++) {
			int u, v, w;
			scanf("%d %d %d", &u, &v, &w);
			// cin >> u >> v >> w;

			if ((dis1[u] + dis2[v] + w) < Min) {
				Min = dis1[u] + dis2[v] + w;
				changeStart = u;
				changeEnd = v;
			}
			else if ((dis1[v] + dis2[u] + w) < Min) {
				Min = dis1[v] + dis2[u] + w;
				changeStart = v;
				changeEnd = u;
			}
		}

		/*cout << "Change Start:" << changeStart << endl;
		cout << "Change End:" << changeEnd << endl;*/

		if (changeStart == 0) {
			Array[0] = E;

			int pre = pre1[E];
			int cnt = 0;

			while (pre != -1) {
				cnt++;
				Array[cnt] = pre;
				pre = pre1[pre];
			}

			for (int i = cnt; i >= 0; i--) {
				printf("%d", Array[i]);
				if (i >= 1) printf(" ");
			}
			printf("\n");

			cout << "Ticket Not Used" << endl;

			printf("%d\n", Min);

		}
		else {
			Array[0] = changeStart;

			int pre = pre1[changeStart];
			int cnt = 0;

			while (pre != -1){
				cnt++;
				Array[cnt] = pre;
				pre = pre1[pre];
			}

			for (int i = cnt; i >= 0; i--) printf("%d ", Array[i]);

			printf("%d", changeEnd);

			pre = pre2[changeEnd];
			while (pre != -1){
				printf(" %d", pre);
				pre = pre2[pre];
			}
			printf("\n");

			printf("%d\n", changeStart);

			printf("%d\n", Min);

		}
	}

	return 0;
}