L3-007 天梯地图 (30 分)（两次dijkstra分别遍历）

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

输入格式：

输入在第一行给出两个正整数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

思路：依次使用两次dijkstra1和dijkstra2分别求：

dijkstra1: 先求最短时间，如果最短时间不唯一，则输出几条最短时间中路径最短的那条

dijkstra2：先求最短路径，如果最短距离的路线不唯一，则输出途径节点数最少的那条

#include<bits/stdc++.h>

using namespace std;

const int N=510,inf=0x3f3f3f3f;

int dist[N],ds[N][N],vis[N],pre1[N];
//dist存储到各点的最短距离，ds存道路的长度，vis标记，pre1存第一条路径的前驱
int times[N],ts[N][N],cnt[N],pre2[N];
//times存储最短时间，ts存储两点之间所需要的时间，cnt存途径节点数，pre2存第二条路径的前驱
int n,m,st,ed;
vector<int>ans1,ans2;//存路径

void dijkstra1(int x)
{
    memset(times,inf,sizeof times);
    memset(vis,0,sizeof vis);
    memset(dist,inf,sizeof dist);
    times[x]=0;
    for(int i=0;i<n;i++)
    {
        int t=-1;
        for(int j=0;j<n;j++)
            if(!vis[j]&&(t==-1||times[j]<times[t]))
                t=j;
        vis[t]=1;
        for(int j=0;j<n;j++)
        {
            if(times[j]>times[t]+ts[t][j])
            {
                times[j]=times[t]+ts[t][j];
                pre1[j]=t;
                dist[j]=dist[t]+ds[t][j];
            }
            else if(times[j]==times[t]+ts[t][j])
            {
                if(dist[j]>dist[t]+ds[t][j])
                {
                    dist[j]=dist[t]+ds[t][j];
                    pre1[j]=t;
                }
            }
        }
    }
}
void dijkstra2(int x)
{
    memset(dist,inf,sizeof inf);
    memset(vis,0,sizeof vis);
    dist[x]=0;
    cnt[x]=1;
    for(int i=0;i<n;i++)
    {
        int t=-1;
        for(int j=0;j<n;j++)
            if(!vis[j]&&(t==-1||dist[t]>dist[j]))
                t=j;
        vis[t]=1;
        for(int j=0;j<n;j++)
        {
            if(dist[j]>dist[t]+ds[t][j])
            {
                dist[j]=dist[t]+ds[t][j];
                pre2[j]=t;
                cnt[j]=cnt[t]+1;
            }
            else if(dist[j]==dist[t]+ds[t][j])
            {
                if(cnt[j]>cnt[t]+1)
                {
                    cnt[j]=cnt[t]+1;
                    pre2[j]=t;
                }
            }
        }
    }
}
int main()
{
    cin>>n>>m;
    memset(ds,inf,sizeof ds);
    memset(ts,inf,sizeof ts);
    while(m--)
    {
        int a,b,op,len,t;
        cin>>a>>b>>op>>len>>t;
        ds[a][b]=len;
        ts[a][b]=t;
        if(!op) ds[b][a]=len,ts[b][a]=t;
    }
    cin>>st>>ed;
    dijkstra1(st);//最快路线中最短的那条
    dijkstra2(st);//途径节点数最少的那条
    int tt=ed;
    while(tt!=st)
    {
        ans1.push_back(tt);
        tt=pre1[tt];
    }
    tt=ed;
    while(tt!=st)
    {
        ans2.push_back(tt);
        tt=pre2[tt];
    }
    int flag=0,lens=max(ans1.size(),ans2.size());
    for(int i=0;i<lens;i++)
        if(ans1[i]!=ans2[i]) flag=1;
    if(flag)//路径不同
    {
        printf("Time = %d: %d",times[ed],st);
        for(int i=ans1.size()-1;i>=0;i--)
            printf(" => %d",ans1[i]);
        cout<<endl;
        printf("Distance = %d: %d",dist[ed],st);
        for(int i=ans2.size()-1;i>=0;i--)
            printf(" => %d",ans2[i]);
        cout<<endl;
    }
    else//路径相同
    {
        printf("Time = %d; Distance = %d: %d",times[ed],dist[ed],st);
        for(int i=ans1.size()-1;i>=0;i--)
            printf(" => %d",ans1[i]);
    }
    return 0;
}