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

求解思路
  • 通过前向星记录边,首先读经济线,读入每条边,应该加两条相反方向的有向边
  • 分别从起点s和终点e各进行一次dijkstra, 从而得到每个点到起点和终点的最短距离,记录在dis1和dis2数组中,当前起点到终点最小值为dis1[e]
  • 读入商业线,起点x,终点y,权值z,每次读入,比较起点到终点最小值 和dis1[x] + dis2[y] + z、dis1[y] + dis2[x] + z,更新最小值,若更新采用该商业线,则将是否选择商业线置为true,并记录该商业线起点、终点、权值。
  • 如果选用了商业线,则将该边加入边池,再进行一次dijkstra
  • 进行dijkstra的过程中,记录前驱结点,从终点寻找前驱结点直到起点,得到最短路径
  • dijkstra算法过程为从起点开始,将当前点距离起点的距离的相反数作为pair的first元素,当前点编号作为pair的second元素,加入priority_queue<pair<int, int>> q;//最大堆。因为pair默认按照first排序,则该堆为距离的最小堆,每次从堆中取出一个离原点最近的元素,如果该点已经到达,则不做处理(很重要,不然堆永远不会弹空),如果未到达,就遍历该点的所有出边,并对边的终点元素进行松弛操作:
for (int i = head[temp.second]; i != -1; i = Edge[i].nxt)
		{
			if (array[Edge[i].v]> array[Edge[i].u] + Edge[i].w)
			{
				array[Edge[i].v] = array[Edge[i].u] + Edge[i].w;
				q.push(make_pair(-array[Edge[i].v], Edge[i].v));
				pre[Edge[i].v] = Edge[i].u;
			}
		}
  • 更新最短距离,记录前驱结点,堆中加入新的点。
  • 有一个优化方式是,如果选了商业边,我还要再进行一次dijkstra,其实如果i在j到k的最短路径上,那么i到k的最短路径一定在j到k的最短路径上。所以可以直接通过pre1找到x到s的边,再通过pre2找y到e的边,注意前驱结点记录的方向,是从商业线向两个端点找。
代码
#include<queue>
#include<vector>
#include <algorithm> 
#include<math.h>
#define inf 1e9
using namespace std;
struct edge
{
	int u=0, v=0, w=0, nxt=0;

};

int dis1[502];
int dis2[502];
int arrive[502];
int pre1[502];
int pre2[502];
//链式前向星
int head[20020];
edge Edge[20020];
int tot = 0;
int n, s, e, m, k;
void addedge(int u, int v, int w)
{
	Edge[tot].u = u;
	Edge[tot].v = v;
	Edge[tot].w = w;
	Edge[tot].nxt = head[u];
	head[u] = tot;
	tot++;
}

void dilkstra(int s, int* array,int *pre)
{
	for (int i = 1; i <= n; i++)//初始化
	{
		arrive[i] = -1;
		array[i] = inf;
		pre[i] = -1;
	}
	priority_queue<pair<int, int>> q;//最大堆 pair.first为距离,pair.second为点
	pair<int, int> temp;
	temp.first = 0; temp.second = s;
	q.push(temp);
	array[s] = 0;
	pre[s] = s;
	while (!q.empty())
	{
		temp = q.top();
		q.pop();
		if (arrive[temp.second]==1) continue;//到达标1
		arrive[temp.second] = 1;
		for (int i = head[temp.second]; i != -1; i = Edge[i].nxt)
		{
			if (array[Edge[i].v]> array[Edge[i].u] + Edge[i].w)
			{
				array[Edge[i].v] = array[Edge[i].u] + Edge[i].w;
				q.push(make_pair(-array[Edge[i].v], Edge[i].v));
				pre[Edge[i].v] = Edge[i].u;
			}
		}
	}
}
int main()
{
	int num = 1;
	while (cin >> n)
	{
		if (num != 1)
		{
			printf("\n");
		}
		num++;
		cin >> s >> e >> m;
		tot = 0;
		for (int i = 1; i <= n; i++)//初始化
		{
			head[i] = -1;
		}
		for (int i = 0; i < 2*m+2; i++)
		{
			Edge[i].u = 0;
			Edge[i].v = 0;
			Edge[i].w = 0;
			Edge[i].nxt = 0;
		}
		for (int i = 0; i < m; i++)
		{
			int x, y, z;
			cin >> x >> y >> z;
			addedge(x, y, z);
			addedge(y, x, z);
		}
		dilkstra(s,dis1, pre1);
		dilkstra(e, dis2, pre2);
		cin >> k;
		int Min = dis1[e];
		bool th = false;//是否经过商业线
		int thx, thy, thz;
		for (int i = 0; i < k; i++)
		{
			int x, y, z;
			cin >> x >> y >> z;
			
			if (Min > dis1[x] + dis2[y] + z)
			{
				Min = dis1[x] + dis2[y] + z;
				th = true;
				thx = x;//记录换乘车站
				thy = y; thz = z;
			}
			if (Min > dis1[y] + dis2[x] + z)//;//啊啊啊不要加 ;
			{
				Min = dis1[y] + dis2[x] + z;
			    th = true;
				thx = y;
				thy = x; thz = z;
			}
		}
	//	cout << thx << thy << thz << endl;
		if (th == true)
		{
			addedge(thx, thy, thz);
			addedge(thy, thx, thz);
			dilkstra(s, dis1, pre1);
		}
		
		vector<int> record;
		
		//if (th == false)
		//{
			int r = e;
			while (pre1[r] != r)
			{
				record.push_back(r);
				r = pre1[r];
			}
			record.push_back(r);
		//}
		
		while (record.size()>1)
		{
			printf("%d ", record.back());
			record.pop_back();
		}
		printf("%d\n", record.back());
		if (th == false)
		{
			printf("Ticket Not Used\n");
		}
		else
		{
			printf("%d\n", thx);
		}
		printf("%d\n", Min);

	}

	return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值