L3-007 天梯地图

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

输入格式：

输入在第一行给出两个正整数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>
#define INF 0x3f3f3f3f
typedef long long ll;
using namespace std;
const int maxn = 5e2 + 10;
int n,m,s,en;//起点，终点
bool vis[maxn];
int dis[maxn];
int pre[maxn];//记录路径
int cnt[maxn];//结点个数
int ss1[maxn];
int ss2[maxn];//路径
struct node
{
    int v,c,di;//相邻结点
    node()
    {
        v = c = di = 0;
    }
    node(int a,int b)
    {
        v = a;
        c = b;

    }
    node(int a,int b,int cc)
    {
        v = a;
        c = b;
        di = cc;
    }
    bool operator<(const node & a)const
    {
        return c > a.c;
    }
};
vector<node> vec1[maxn];//距离
vector<node> vec2[maxn];//时间
priority_queue<node> que;
void addEdge1(int u,int v,int w,int x)
{
    vec1[u].push_back(node(v,w,x));
}
void addEdge2(int u,int v,int w,int x)
{
    vec2[u].push_back(node(v,w,x));
}
bool judge(int a,int b)
{
    if(a != b) return false;
    for(int i = 0;i < a;i++)
    {
        if(ss1[i] != ss2[i]) return false;
    }
    return true;
}
void dijkstra2()
{
    memset(vis,0,sizeof(vis));
    for(int i = 0;i < n;i++)
    {
        pre[i] = i;
        dis[i] = INF;
        cnt[i] = INF;
    }
    while(!que.empty()) que.pop();
    pre[s] = -1,dis[s] = cnt[s] = 0;
    que.push(node(s,0));
    node tmp;
    while(!que.empty())
    {
        tmp = que.top(); que.pop();
        int u = tmp.v;
        if(vis[u]) continue;
        vis[u] = true;
        int l = vec2[u].size();
        for(int i = 0;i < l;i++)
        {
            int v = vec2[u][i].v;
            int cost = vec2[u][i].c;
            int sec = vec2[u][i].di;
            if(!vis[v] && dis[v] > dis[u] + cost)
            {
                pre[v] = u;
                cnt[v] = cnt[u] + sec;
                dis[v] = dis[u] + cost;
                que.push(node(v,dis[v]));
            }
            else if(!vis[v] && dis[v] == dis[u] + cost)
            {
                if(cnt[v] > cnt[u] + sec)
                {
                    pre[v] = u;
                    cnt[v] = cnt[u] + sec;
                }
            }
        }
    }
}
void dijkstra1()
{
    memset(vis,0,sizeof(vis));
    for(int i = 0;i < n;i++)
    {
        pre[i] = i;
        dis[i] = INF;
        cnt[i] = INF;
    }
    while(!que.empty()) que.pop();

    pre[s] = -1,dis[s] = 0,cnt[s] = 1;
    que.push(node(s,0));
    node tmp;
    while(!que.empty())
    {
        tmp = que.top(); que.pop();
        int u = tmp.v;
        if(vis[u]) continue;
        vis[u] = true;
        int l = vec1[u].size();
        for(int i = 0;i < l;i++)
        {
            int v = vec1[u][i].v;
            int cost = vec1[u][i].c;
            if(!vis[v] && dis[v] > dis[u] + cost)
            {
                pre[v] = u;
                cnt[v] = cnt[u] + 1;
                dis[v] = dis[u] + cost;
                que.push(node(v,dis[v]));
            }
            else if(!vis[v] && dis[v] == dis[u] + cost)
            {
                if(cnt[v] > cnt[u] + 1)
                {
                    pre[v] = u;
                    cnt[v] = cnt[u] + 1;
                    que.push(node(v,dis[v]));
                }
            }
        }
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    int a,b,c,d,e;//结点1，结点2，是否单行，距离，时间
    for(int i = 0;i < m;i++)
    {
        scanf("%d%d%d%d%d",&a,&b,&c,&d,&e);
        if(c == 0)
        {
            addEdge1(b,a,d,e);
            addEdge2(b,a,e,d);
        }
        addEdge1(a,b,d,e);
        addEdge2(a,b,e,d);
    }
    scanf("%d%d",&s,&en);
    dijkstra2();//时间
    int now = en;
    int pos1 = 0,pos2 = 0;
    int dis1 = dis[en];
    while(pre[now] != -1)
    {
        ss1[pos1++] = now;
        now = pre[now];
    }
    dijkstra1();//距离
    now = en;
    while(pre[now] != -1)
    {
        ss2[pos2++] = now;
        now = pre[now];
    }
    if(judge(pos1,pos2))
    {
        printf("Time = %d; Distance = %d: %d",dis1,dis[en],s);
        for(int i = pos2 - 1;i>=0;i--)
        {
            printf(" => %d",ss2[i]);
        }
        printf("\n");
        return 0;
    }
    printf("Time = %d: %d",dis1,s);
    for(int i = pos1 - 1;i>=0;i--)
    {
        printf(" => %d",ss1[i]);
    }
    printf("\n");
    printf("Distance = %d: %d",dis[en],s);
    for(int i = pos2 - 1;i>=0;i--)
    {
        printf(" => %d",ss2[i]);
    }
    printf("\n");
    return 0;
}