L3-007 天梯地图 - 最短路

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

思路：

千万不要：先求出最短距离，再在求最短时间的时候用到最短距离，因为你先求出的最短距离，再求最短时间的时候，走的可能不是对应的最短距离的那条路。

求最短时间：

开一个juli向量记录，最短时间到点i时的经过的路径的距离，然后遇到相同的路径，比较juli向量。

求最短距离：

如果最短距离的路线不唯一，则输出途径节点数最少的那条，可以再开一个数组记录最短的节点数（把它看成是距离为1的路径，求最短路）

代码如下：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<queue>
#include<stack>
#include<cmath>
#include<set>
#include<map>
using namespace std;
#define ll long long
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define ull unsigned long long
typedef pair<int,int>P;
const int INF=0x3f3f3f3f;
const int N=515,mod=32767;

int costL[N][N],costT[N][N],vis1[N],vis2[N],d1[N],d2[N];
int juli[N];

int n,S,T,pre1[N],pre2[N],cnt[N];
vector<int>ans1,ans2;
void init(){
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            if(i!=j){
                costL[i][j]=INF;
                costT[i][j]=INF;
            }
        }
    }
}

void dijk(){
    memset(vis1,0,sizeof(vis1));
    memset(vis2,0,sizeof(vis2));
    memset(pre1,-1,sizeof(pre1));
    memset(pre2,-1,sizeof(pre2));
    for(int i=0;i<n;i++){
        d1[i]=INF;
        d2[i]=INF;
        cnt[i]=INF;
    }
    d1[S]=0;d2[S]=0;
    juli[S]=0;
    cnt[S]=0;
    int v1,v2;
    for(int j=0;j<n;j++){
        v1=-1;
        for(int i=0;i<n;i++){
            if(!vis1[i]&&(v1==-1||d1[i]<d1[v1]))v1=i;
        }
        vis1[v1]=1;
        for(int i=0;i<n;i++){
            if(i==v1||vis1[i])continue;
            if(d1[i]>d1[v1]+costT[v1][i]){
                pre1[i]=v1;
                d1[i]=d1[v1]+costT[v1][i];
                juli[i]=juli[v1]+costL[v1][i];
            }
            else if(d1[i]==d1[v1]+costT[v1][i]){
                if(juli[i]>juli[v1]+costL[v1][i]){
                    pre1[i]=v1;
                }
            }
        }
        v2=-1;
        for(int i=0;i<n;i++){
            if(!vis2[i]&&(v2==-1||d2[i]<d2[v2]))v2=i;
        }
        vis2[v2]=1;
        //cnt[v2]=j;
        for(int i=0;i<n;i++){
            if(i==v2||vis2[i])continue;
            if(d2[i]>d2[v2]+costL[v2][i]){
                d2[i]=d2[v2]+costL[v2][i];
                cnt[i]=cnt[v2]+1;
                pre2[i]=v2;
            }
            else if(d2[i]==d2[v2]+costL[v2][i]){
                if(cnt[i]>cnt[v2]+1){
                    cnt[i]=cnt[v2]+1;
                    pre2[i]=v2;
                }
            }
        }
    }
}

void print(int flag,int d[],vector<int>ans){
    if(flag==1)printf("Time = %d: ",d[T]);
    else printf("Distance = %d: ",d[T]);
    printf("%d",ans[ans.size()-1]);
    for(int i=ans.size()-2;i>=0;i--){
        printf(" => %d",ans[i]);
    }
    printf("\n");
}

int main(){
    int m,v1,v2,flag,len,time;
    scanf("%d%d",&n,&m);
    init();
    for(int i=1;i<=m;i++){
        scanf("%d%d%d%d%d",&v1,&v2,&flag,&len,&time);
        costL[v1][v2]=len;
        costT[v1][v2]=time;
        if(!flag){
            costL[v2][v1]=len;
            costT[v2][v1]=time;
        }
    }
    scanf("%d%d",&S,&T);
    dijk();
    ans1.push_back(T);
    for(int i=pre1[T];i!=-1;i=pre1[i]){
        ans1.push_back(i);
    }
    ans2.push_back(T);
    for(int i=pre2[T];i!=-1;i=pre2[i]){
        ans2.push_back(i);
    }

    int f=0;
    if((ans1.size())==(ans2.size())){
        int f1=0;
        for(int i=0;i<ans1.size();i++){
            if(ans1[i]!=ans2[i]){
                f1=1;
                break;
            }
        }
        if(f1==0)f=1;
    }
    if(f){
        printf("Time = %d; Distance = %d: ",d1[T],d2[T]);
        printf("%d",ans1[ans1.size()-1]);
        for(int i=ans1.size()-2;i>=0;i--){
            printf(" => %d",ans1[i]);
        }
        printf("\n");
    }
    else {
        print(1,d1,ans1);
        print(2,d2,ans2);
    }
}