（最短路问题）L3-007 天梯地图 (30 分)

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 <iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<stack>
using namespace std;
int mdis[510][510];  //用来存储图中路径
int mtim[510][510];   //用来存储图中时间
int vis[510];     //用来标记是否已经加入到集合
int tim[510];       //用来储存当前起点到当前点最短时间
int dis[510];     //用来储存起点到当前点的最短路径
int p1[510];     //用来储存当前节点的上一个节点
int p2[510];
int cnt[510];   //用来储存走过的节点数
int n,m;    //节点数与路径数
const int inf=0x3f3f3f3f;
void init()   //初始化函数
{
    for(int i=0;i<510;i++)
    {
        for(int j=0;j<510;j++)
        {
            if(i!=j)
            {
                mdis[i][j]=inf;
                mtim[i][j]=inf;
            }
            else
            {
                mdis[i][j]=0;
                mtim[i][j]=0;
            }
        }
    }
}
void dij1(int s,int t)  //计算最短路径
{
    memset(vis,0,sizeof(vis));
    memset(cnt,1,sizeof(cnt));
    for(int i=0;i<n;i++)
    {
        dis[i]=mdis[s][i];
        p1[i]=s;
    }
    vis[s]=1;
    p1[s]=-1;
    for(int i=1;i<n;i++)
    {
        int temp=inf,k;
        for(int j=0;j<n;j++)  //找到当前最近点加入集合
        {
            if(!vis[j]&&temp>dis[j])
            {
                temp=dis[j];
                k=j;
            }
        }
        vis[k]=1;
        if(k==t||temp==inf)  //当找到终点或无路可走时返回
            return;
        for(int j=0;j<n;j++)  //更新路径
        {
            if(!vis[j]&&dis[j]>dis[k]+mdis[k][j])
            {
                dis[j]=dis[k]+mdis[k][j];
                cnt[j]=cnt[k]+1;
                p1[j]=k;
            }
            else if(!vis[j]&&dis[j]==dis[k]+mdis[k][j]&&cnt[j]>cnt[k]+1)
            {
                cnt[j]=cnt[k]+1;
                p1[j]=k;
            }
        }
    }
}

void pr(int t)  //输出函数
{
    stack<int> q1;
    stack<int> q2;
    int aa=t;
    while(p1[aa]!=-1)
    {
        q1.push(aa);
        aa=p1[aa];
    }
    q1.push(aa);
    aa=t;
    while(p2[aa]!=-1)
    {
        q2.push(aa);
        aa=p2[aa];
    }
    q2.push(aa);
    if(q1==q2)
    {
        printf("Time = %d; Distance = %d:",tim[t],dis[t]);
        while(!q1.empty())
        {
            int bb=q1.top();
            q1.pop();
            printf(" %d",bb);
            if(bb!=t)
                printf(" =>");
        }
        printf("\n");
    }
    else
    {
        printf("Time = %d:",tim[t]);
        while(!q2.empty())
        {
            int bb=q2.top();
            q2.pop();
            printf(" %d",bb);
            if(bb!=t)
                printf(" =>");
        }
        printf("\n");
        printf("Distance = %d:",dis[t]);
        while(!q1.empty())
        {
            int bb=q1.top();
            q1.pop();
            printf(" %d",bb);
            if(bb!=t)
                printf(" =>");
        }
        printf("\n");
    }
}
void dij2(int s,int t)  //计算最短时间
{
    memset(vis,0,sizeof(vis));
    memset(cnt,1,sizeof(cnt));
    for(int i=0;i<n;i++)
    {
        dis[i]=mdis[s][i];
        tim[i]=mtim[s][i];
        p2[i]=s;
    }
    vis[s]=1;
    p2[s]=-1;
    for(int i=1;i<n;i++)
    {
        int temp=inf,k;
        for(int j=0;j<n;j++)
        {
            if(!vis[j]&&temp>tim[j])
            {
                temp=tim[j];
                k=j;
            }
        }
        vis[k]=1;
        if(k==t||temp==inf)
            return ;
        for(int j=0;j<n;j++)
        {
            if(!vis[j])
            {
                if(tim[j]>tim[k]+mtim[k][j])
                {
                    tim[j]=tim[k]+mtim[k][j];
                    dis[j]=dis[k]+mdis[k][j];
                    p2[j]=k;
                }
                else if(tim[j]==tim[k]+mdis[k][j]&&dis[j]>dis[k]+mdis[k][j])
                {
                    dis[j]=dis[k]+mdis[k][j];
                    p2[j]=k;
                }
            }
        }
    }
}
int main()
{
    cin>>n>>m;
    init();
    for(int i=0;i<m;i++)
    {
        int v1,v2,f,len,tt;
        cin>>v1>>v2>>f>>len>>tt;
        mdis[v1][v2]=len;
        mtim[v1][v2]=tt;
        if(!f)
        {
            mdis[v2][v1]=len;
            mtim[v2][v1]=tt;
        }
    }
    int s,t;
    cin>>s>>t;
    dij2(s,t);  //要先2 再 1因为dis数组用的同一个
    dij1(s,t);
    pr(t);
    return 0;
}