week7作业B TT旅行日记 Dijkstra

题目描述

众所周知，TT 有一只魔法猫。
今天他在 B 站上开启了一次旅行直播，记录他与魔法猫在喵星旅游时的奇遇。 TT 从家里出发，准备乘坐猫猫快线前往喵星机场。猫猫快线分为经济线和商业线两种，它们的速度与价钱都不同。当然啦，商业线要比经济线贵，TT 平常只能坐经济线，但是今天 TT 的魔法猫变出了一张商业线车票，可以坐一站商业线。假设 TT 换乘的时间忽略不计，请你帮 TT 找到一条去喵星机场最快的线路，不然就要误机了！
Input
输入包含多组数据。每组数据第一行为 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 行是商业线路段的描述，格式同经济线。
所有路段都是双向的，但有可能必须使用商业车票才能到达机场。保证最优解唯一。
Output
对于每组数据，输出3行。第一行按访问顺序给出 TT 经过的各个车站（包括起点和终点），第二行是 TT 换乘商业线的车站编号（如果没有使用商业线车票，输出"Ticket Not Used"，不含引号），第三行是 TT 前往喵星机场花费的总时间。
本题不忽略多余的空格和制表符，且每一组答案间要输出一个换行。

样例

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

思路分析

本题中用到了dijkstra算法，我采用的方法是，先分别以起点和终点为源点求出单源最短路dis1、dis2数组，得到不走商业线的最短路答案，再枚举每条商业线（u,v,w）计算起点到u的最短路+w+w到终点的最短路，最后取最小值。

Dijkstra算法简述：这里最小堆用优先级队列实现，并创建heap类。

struct heap
{
 int u,d;
 bool operator<(const heap &p) const
 {
  return d>p.d;
 }
};
void dijkstra(int s,int *dis,int *pre)
{
 priority_queue<heap> q;
 heap cur;
 cur.d=0;cur.u=s;
 q.push(cur);
 dis[s]=0;
 pre[s]=s;
 while(!q.empty())
 {
  cur=q.top();q.pop();
  int u=cur.u;
  if(vis[u]==1) continue;
  vis[u]=1;
  for(int i=head[u];~i;i=edge[i].next)
  {
   int v=edge[i].v;
   if(dis[v]>dis[u]+edge[i].w)
   {
    dis[v]=dis[u]+edge[i].w;
    pre[v]=u;
    heap temp;
    temp.d=dis[v];temp.u=v;
    q.push(temp);
   }
  }
 }
}

用pre1数组，pre2数组来记录两次Dijkstr中的前驱点，输出时定义vector数组来接收从起点到终点正确的输出数组。

AC代码

#include<cstdio> 
#include<cstring>
#include<queue>
#include<cmath>
#include<vector>
using namespace std;
const int inf=5*1e8;
int n,S,E;
struct Edge
{
 int v,w,next;//相邻的点，权值，下条边的序号 
} ;
int head[600],vis[600],dis1[600],dis2[600],pre1[600],pre2[600];
int tot;//Edge下标
Edge edge[20000];
void init()
{
 //初始化 
 tot=0;
 memset(head,-1,sizeof(head));
 memset(vis,0,sizeof(vis));
 memset(pre1,0,sizeof(pre1));
 memset(pre2,0,sizeof(pre2));
 for(int i=0;i<=n;i++)
 {
  dis1[i]=inf;
  dis2[i]=inf;
 }
 } 
void add(int u,int v,int w)
{
 edge[tot].v=v;
 edge[tot].w=w;
 edge[tot].next=head[u];
 head[u]=tot;
 tot++;
}
struct heap
{
 int u,d;
 bool operator<(const heap &p) const
 {
  return d>p.d;
 }
};
void dijkstra(int s,int *dis,int *pre)
{
 priority_queue<heap> q;
 heap cur;
 cur.d=0;cur.u=s;
 q.push(cur);
 dis[s]=0;
 pre[s]=s;
 while(!q.empty())
 {
  cur=q.top();q.pop();
  int u=cur.u;
  if(vis[u]==1) continue;
  vis[u]=1;
  for(int i=head[u];~i;i=edge[i].next)
  {
   int v=edge[i].v;
   if(dis[v]>dis[u]+edge[i].w)
   {
    dis[v]=dis[u]+edge[i].w;
    pre[v]=u;
    heap temp;
    temp.d=dis[v];temp.u=v;
    q.push(temp);
   }
  }
 }
}
int main()
{
 int flag=0;
 while(scanf("%d%d%d",&n,&S,&E)!=EOF)
 {
  if(flag!=0) printf("\n");
     init();
  int m,x,y,z;
  scanf("%d",&m);
  for(int i=0;i<m;i++)
  {
   scanf("%d%d%d",&x,&y,&z);
   add(x,y,z);
   add(y,x,z);
  }
  dijkstra(S,dis1,pre1);
  memset(vis,0,sizeof(vis));
  dijkstra(E,dis2,pre2);
  int k;
  scanf("%d",&k);
  int u=0,v=0,ans=dis1[E];
  for(int i=0;i<k;i++)//商业线 
  {
   scanf("%d%d%d",&x,&y,&z);
   if((dis1[x]+dis2[y]+z)<(dis2[x]+dis1[y]+z))
   {
    if(dis1[x]+dis2[y]+z<ans)//使用商业线 
    {
     ans=dis1[x]+dis2[y]+z;
     u=x;v=y;
    }
   }
   else
   {
    if(dis2[x]+dis1[y]+z<ans)
    {
     ans=dis2[x]+dis1[y]+z;
     u=y;v=x;
    }
   } 
  }
  //cout<<u<<" "<<ans<<endl;
  vector<int> path;
  if(u==0&&v==0)//不使用商业线 
  {
   for(int i=E;i!=S;i=pre1[i]) 
   {
    path.push_back(i);
   }
   path.push_back(S);
   for(int i=path.size()-1;i>=1;i--)
   {
    printf("%d ",path[i]);
   }
   printf("%d\n",path[0]);
   printf("Ticket Not Used\n%d\n",ans);
   } 
   else
   {
    for(int i=u;i!=S;i=pre1[i]) 
   {
    path.push_back(i); 
   }
   path.push_back(S);
   for(int i=v;i!=E;i=pre2[i]) 
   {
    path.insert(path.begin(),i); 
   }
   path.insert(path.begin(),E);
   for(int i=path.size()-1;i>=1;i--)
   {
    printf("%d ",path[i]);
   }
   printf("%d\n",path[0]);
   printf("%d\n%d\n",u,ans);
   }
   flag++;
 }
 return 0;
}