天梯地图

Posted 2021-02-13 czy-power

L3-007 天梯地图 （30 分)

本题要求你实现一个天梯赛专属在线地图，队员输入自己学校所在地和赛场地点后，该地图应该推荐两条路线：一条是最快到达路线；一条是最短距离的路线。题目保证对任意的查询请求，地图上都至少存在一条可达路线。

输入格式：

输入在第一行给出两个正整数N（2 ≤ N ≤ 500）和M，分别为地图中所有标记地点的个数和连接地点的道路条数。随后M行，每行按如下格式给出一条道路的信息：

V1 V2 one-way length time

其中V1和V2是道路的两个端点的编号（从0到N-1）；如果该道路是从V1到V2的单行线，则one-way为1，否则为0；length是道路的长度；time是通过该路所需要的时间。最后给出一对起点和终点的编号。

输出格式：

首先按下列格式输出最快到达的时间T和用节点编号表示的路线：

Time = T: 起点 => 节点1 => ... => 终点

然后在下一行按下列格式输出最短距离D和用节点编号表示的路线：

Distance = D: 起点 => 节点1 => ... => 终点

如果最快到达路线不唯一，则输出几条最快路线中最短的那条，题目保证这条路线是唯一的。而如果最短距离的路线不唯一，则输出途径节点数最少的那条，题目保证这条路线是唯一的。

如果这两条路线是完全一样的，则按下列格式输出：

Time = T; Distance = D: 起点 => 节点1 => ... => 终点

输入样例1：

10 15
0 1 0 1 1
8 0 0 1 1
4 8 1 1 1
5 4 0 2 3
5 9 1 1 4
0 6 0 1 1
7 3 1 1 2
8 3 1 1 2
2 5 0 2 2
2 1 1 1 1
1 5 0 1 3
1 4 0 1 1
9 7 1 1 3
3 1 0 2 5
6 3 1 2 1
5 3

输出样例1：

Time = 6: 5 => 4 => 8 => 3
Distance = 3: 5 => 1 => 3

输入样例2：

7 9
0 4 1 1 1
1 6 1 3 1
2 6 1 1 1
2 5 1 2 2
3 0 0 1 1
3 1 1 3 1
3 2 1 2 1
4 5 0 2 2
6 5 1 2 1
3 5

输出样例2：

Time = 3; Distance = 4: 3 => 2 => 5

#include <bits/stdc++.h>
#include<queue>

using namespace std;
const int maxn=1e6+5;
typedef pair<int,int> P;
int n,m;
vector<int>ans1,ans2;
struct node{
     int to,l,t;
};
vector<node>p[maxn];
int d1[maxn],d2[maxn],tmm[maxn],p1[maxn],p2[maxn],s,t,tot[maxn];
void dijtm(){
      priority_queue<P,vector<P>,greater<P> >q;
      memset(d2,127,sizeof(d2));
      memset(tmm,127,sizeof(tmm));
      q.push(P(s,0));
      tmm[s]=0;
      d2[s]=0;
      while(q.size()){
           P cur=q.top();
           q.pop();
           int nd=cur.first,tmptm=cur.second;
           if(tmm[nd]<tmptm)continue;
           for(int i=0;i<p[nd].size();i++){
                int to=p[nd][i].to;
                int tmpdis2=p[nd][i].l;
                int tmptm2=p[nd][i].t;
                if(tmm[to]>tmm[nd]+tmptm2){
                    tmm[to]=tmm[nd]+tmptm2;
                    q.push(P(to,tmptm2+tmm[nd]));
                    d2[to]=d2[nd]+p[nd][i].l;
                    p2[to]=nd;
                }
                else if(tmm[to]==tmm[nd]+tmptm2){
                      if(d2[to]>d2[nd]+tmpdis2){
                           d2[to]=d2[nd]+tmpdis2;
                           p2[to]=nd;
                      }
                }
           }
      }
      return;
}
void dijdis(){
      priority_queue<P,vector<P>,greater<P> >q;
      memset(d1,127,sizeof(d1));
      q.push(P(s,0));
      d1[s]=0;
      while(q.size()){
            P cur=q.top();
            q.pop();
            int nd=cur.first,tmpdis=cur.second;
            if(d1[nd]<tmpdis)continue;
            for(int i=0;i<p[nd].size();i++){
                   int to=p[nd][i].to;
                   int tmpdis2=d1[nd]+p[nd][i].l;
                   if(d1[to]>tmpdis2){
                        d1[to]=tmpdis2;
                        tot[to]=tot[nd]+1;
                        q.push(P(to,d1[to]));
                        p1[to]=nd;
                   }
                   else if(tmpdis2==d1[to]){
                         if(tot[to]>tot[nd]+1){
                            tot[to]=tot[nd]+1;
                            p1[to]=nd;
                         }
                   }
            }
      }
      return;
}
void dfs(int *tmp,int cur,vector<int>& vtmp){
       if(cur==s){
           vtmp.push_back(cur);
           return;
       }
       dfs(tmp,tmp[cur],vtmp);
       vtmp.push_back(cur);
       return;
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=m;i++){
        int v1,v2,o,len,t;
        scanf("%d%d%d%d%d",&v1,&v2,&o,&len,&t);
        if(o)p[v1].push_back(node{v2,len,t});
        else {
             p[v1].push_back(node{v2,len,t});
             p[v2].push_back(node{v1,len,t});
        }
    }
    cin>>s>>t;
    dijtm();
    dijdis();
    dfs(p1,t,ans1);
    dfs(p2,t,ans2);
    bool flag=false;
    if(ans1.size()!=ans2.size())flag=true;
    else{
        for(int i=0;i<ans1.size();i++){
             if(ans1[i]!=ans2[i]){
                 flag=true;
                 break;
             }
        }
    }
    if(!flag){
        printf("Time = %d; Distance = %d: %d",tmm[t],d1[t],s);
        for(int i=1;i<ans1.size();++i)printf(" => %d",ans1[i]);
    }
    else{
        printf("Time = %d: %d",tmm[t],s);
        for(int i=1;i<ans2.size();++i)printf(" => %d",ans2[i]);
        printf("
Distance = %d: %d",d1[t],s);
        for(int i=1;i<ans1.size();++i)printf(" => %d",ans1[i]);
    }
    return 0;
}