WEEK7 周记 作业——Dijkstra算法_TT的旅行日记

WEEK7 周记 作业——Dijkstra算法_TT的旅行日记

一、题意

1.简述

众所周知，TT 有一只魔法猫。
今天他在 B 站上开启了一次旅行直播，记录他与魔法猫在喵星旅游时的奇遇。 TT 从家里出发，准备乘坐猫猫快线前往喵星机场。猫猫快线分为经济线和商业线两种，它们的速度与价钱都不同。当然啦，商业线要比经济线贵，TT 平常只能坐经济线，但是今天 TT 的魔法猫变出了一张商业线车票，可以坐一站商业线。假设 TT 换乘的时间忽略不计，请你帮 TT 找到一条去喵星机场最快的线路，不然就要误机了！

2.输入格式

输入包含多组数据。每组数据第一行为 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.输出格式

对于每组数据，输出3行。第一行按访问顺序给出 TT 经过的各个车站（包括起点和终点），第二行是 TT 换乘商业线的车站编号（如果没有使用商业线车票，输出"Ticket Not Used"，不含引号），第三行是 TT 前往喵星机场花费的总时间。

本题不忽略多余的空格和制表符，且每一组答案间要输出一个换行

4.样例

Input

4 1 4
4
1 2 2
1 3 3
2 4 4
3 4 5
1
2 4 3

Output

1 2 4
2
5

二、算法

主要思路

题目给定了起点与终点，而且要求商业线最多乘坐一次。可以枚举每一条商业线，计算起点到u的最短路以及v到终点的最短路再加上该商业线所花费的时间。以起点为源点求单源最短路，得到 $dis1$ 数组，再以终点为源点求单源最短路，得到 $dis2$ 数组枚举商业线(u, v, w)，取 m i n { d i s 1 [ u ] + d i s 2 [ v ] + w , d i s 1 [ v ] + d i s 2 [ u ] + w } min\{dis1[u]+dis2[v]+w,dis1[v]+dis2[u]+w \} min{dis1[u]+dis2[v]+w,dis1[v]+dis2[u]+w}，最终再与不走商业线的答案取$min$。

---

三、代码

#include<iostream>
#include<cstring>
#include<string>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<queue>
using namespace std;
struct D{
	int dis,st,pre;
	bool operator<(const D& d)const{
		return dis>d.dis;
	}
};

struct Edge{
	int a,b,lth;
	Edge(int a=0,int b=0,int lth=0):a(a),b(b),lth(lth){}
};
D dis1[510];
D dis2[510];
int vis[510];
int v[510][510];
Edge ed[1010];
int line[510];
int n;
void print(int st,int& index,int line[]){
	if(dis1[st].pre == st) return;
	print(dis1[st].pre,index,line);
	line[index++] = dis1[st].pre;
}
void print_reverse(int st,int& index,int line[]){
	if(dis2[st].pre == st) return;
	line[index++] = st;
	print_reverse(dis2[st].pre,index,line);
}
void dijstra(int s,D dis[]){
	memset(vis,0,sizeof(vis));
	dis[s].dis = 0;
	priority_queue<D> q;
	q.push(dis[s]);
	while(!q.empty()){
		D d = q.top();
		q.pop();
		if(vis[d.st] == 1) 
			continue;
		vis[d.st] = 1;
		for(int i=1;i<=n;i++){
			if(i==d.st)
				continue;
			if(v[d.st][i]!=0){//是临界点 
				if(v[d.st][i]+d.dis < dis[i].dis){
					dis[i].dis = v[d.st][i]+d.dis;
					dis[i].pre = d.st;
					q.push(dis[i]);
				}	
			}
		}
	}
}
int main()
{
	int s,e;
	int cnt = 0;
	while(scanf("%d%d%d",&n,&s,&e)!=EOF){ 
		cnt++;
		memset(line,0,sizeof(line));
		memset(ed,0,sizeof(ed));//0代表没有连边 
		memset(v,0,sizeof(v));
		memset(vis,0,sizeof(vis));
		memset(dis1,63,sizeof(dis1));
		memset(dis2,63,sizeof(dis2));
		for(int i=1;i<=n;i++){
			dis1[i].pre = i;
			dis2[i].pre = i;
			dis1[i].st = i;
			dis2[i].st = i;
		}
		int m,k;
		scanf("%d",&m);
		for(int i=0;i<m;i++){
			int x,y,z;
			scanf("%d%d%d",&x,&y,&z);
			v[x][y] = z;
			v[y][x] = z;
		}
		scanf("%d",&k);
		int edindex = 0; 
		for(int i=0;i<k;i++){
			int x,y,z;
			scanf("%d%d%d",&x,&y,&z);
			ed[edindex++] = Edge(x,y,z);
		}
		dijstra(s,dis1);
		dijstra(e,dis2);
		
		int mindis = dis1[e].dis;
		int cst = -1;
		int cend = -1;
		for(int i=0;i<edindex;i++){
			if(mindis>dis1[ed[i].a].dis+dis2[ed[i].b].dis+ed[i].lth){
				mindis = dis1[ed[i].a].dis+dis2[ed[i].b].dis+ed[i].lth;
				cst = ed[i].a;
				cend = ed[i].b;
			}
			if(mindis>dis2[ed[i].a].dis+dis1[ed[i].b].dis+ed[i].lth){
				mindis = dis2[ed[i].a].dis+dis1[ed[i].b].dis+ed[i].lth;
				cst = ed[i].b;
				cend = ed[i].a;
			}
		}
		int index = 0;
		if(cst != -1){//经历了商业线 
			print(cst,index,line);
			line[index++] = cst;
			print_reverse(cend,index,line);
			line[index++] = e;
		}
		else{
			print(e,index,line);
			line[index++] = e;
		}
		if(cnt>1)
			printf("\n");
		for(int i=0;i<index;i++){
			if(i>0) printf(" ");
			printf("%d",line[i]);
		}
		printf("\n");
		if(cst != -1)
			printf("%d\n",cst);
		else 
		    printf("Ticket Not Used\n");
		printf("%d\n",mindis);
	}
	return 0;
}