dijkstral算法——线路选择问题

问题描述

众所周知，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算法： 用于解决图中没有负边的单源最短路问题
流程：
设置 s 为源点,dis【a】 表示源点 s 到点 a 的最短距离, 初
始化,dis【s】 = 0, ,dis【i】 = inf (无穷大), 将 s 加入最小堆
• 每次从堆中取出一个点 x，遍历 x 的所有邻接边
(x，y，w) ，比较 ,dis【y】 与,dis【x】 + w 的大小，松弛操作
松弛操作： 比较 dis【y】 与 dis【x】 + w 的大小这一过程称为松弛操作
• 若 dis【y】 > dis【x】 + w，则松弛成功
• 更新dis【y】 的大小
• 将 y 加入最小堆中
• 不断执行上述操作，直至最小堆为空，最后得到的 dis 数组
即为所求单源最短路值
• 注意
• 由于图中只有正边，因此每个点只会被最小堆弹出一次
• 即一旦某个点被最小堆弹出，则不会再被松弛，dis 的值
为最短路
每次弹出堆中最短边，然后遍历最短边的点进行松弛：当松弛边 (u, v) 时，如果 dis[u] 已经是最短路且 u 在 v 的最短
路上，则松弛结束后 dis[v] 也是最短路，并且从此以后其 dis 值不会发生变化

本题思路：
1、以起点为源点求单源最短路，得到 dis1 数组
2、再以终点为源点求单源最短路，得到 dis2 数组
3、 枚举商业线(u, v, w)，取 min{dis1[u]+dis2[v]+w,
dis1[v]+dis2[u]+w}
4、与不走商业线只走经济线的答案取min

#include<iostream>
#include<queue>
#include<cstring>
#include<cstdio> 
const int inf=5*1e8;
using namespace std;

struct edge
{
	int u,v,w,nxt;
}edges[510*510];

int head[1010],tot=1;
int N,S,E,K,M,l,r; 
void addEdge(int u,int v,int w)
{
	edges[tot].u=u;
	edges[tot].v=v;
	edges[tot].w=w;
	edges[tot].nxt=head[u];
	head[u]=tot;
	tot++;
}

int vis[1010];
int dis1[1010],dis2[1010],pre1[1010],pre2[1010];
priority_queue<pair<int ,int> > q;
void dijkstral1(int s)
{
	while(q.size()) 
		q.pop();//清空堆
	memset(vis,0,sizeof(vis));
	for(int i=1;i<=N;i++)
	    dis1[i]=inf;//距离初始化为一个很大的数
	dis1[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=edges[i].nxt)
		{
			int y=edges[i].v;
			int w=edges[i].w;
			if(dis1[y]>dis1[x]+w)
			{//松弛操作 
			    pre1[y]=x; //记录路径 
				dis1[y]=dis1[x]+w;
				q.push(make_pair(-dis1[y],y));
			}
		} 
	} 
}

void dijkstral2(int s)
{
	while(q.size()) 
		q.pop();//清空堆
	memset(vis,0,sizeof(vis));
	for(int i=1;i<=N;i++)
	    dis2[i]=inf;//距离初始化为一个很大的数
	dis2[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=edges[i].nxt)
		{
			int y=edges[i].v;
			int w=edges[i].w;
			if(dis2[y]>dis2[x]+w)
			{//松弛操作 
			    pre2[y]=x; //记录路径 
				dis2[y]=dis2[x]+w;
				q.push(make_pair(-dis2[y],y));
			}
		} 
	} 
}
//从起点到u点
void out1(int u)
{
	if(u==S)
	{
		cout<<u;
		return;
	}
	out1(pre1[u]);
	cout<<" "<<u;
}
//从u点到终点
void out2(int u)
{
	cout<<" "<<u;
	if(u==E)  return;
	out2(pre2[u]);
	 
} 
 
int main()
{
	bool flag=false;
	while(cin>>N>>S>>E)//车站总数，起点，终点 
	{
		if(flag) cout<<endl;
		memset(head,-1,sizeof(head));
		memset(pre1,-1,sizeof(pre1));
		memset(pre2,-1,sizeof(pre2));
		tot=1;
		l=r=0;
		cin>>M;
		int u,v,w;
		for(int i=0;i<M;i++)
		{
			cin>>u>>v>>w;
			addEdge(u,v,w);
			addEdge(v,u,w);//无向图 
		}
		dijkstral1(S);//距起点的最短路径 
		dijkstral2(E);//距终点的最短路径
		int minw=dis1[E];//只坐经济线 
	    cin>>K; 
	    for(int i=1;i<=K;i++)
	    {//枚举每一条商业线，起点不确定 
	    	cin>>u>>v>>w;
	    	int temp1=dis1[u]+dis2[v]+w;
			int temp2=dis1[v]+dis2[u]+w;
			minw=min(min(temp1,temp2),minw);
			if(minw==temp1)
			{
				l=u;r=v;
			} 
			else if(minw==temp2)
			{
				l=v;r=u;
			}
		}
		if(l==0&&r==0)//只经济线
		{
			out1(E);
			cout<<endl<<"Ticket Not Used"<<endl<<minw<<endl;
		} 
		else
		{
			out1(l);
			out2(r);//getchar从标准输入里读取下一个字符，不能解决输出 
			cout<<endl<<l<<endl<<minw<<endl;
		}
		flag=true;
	} 
}