7-4 天梯地图 （30 分）

 

 

7-4 天梯地图 （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

 

PS:突然发现nlogn的dijkstra也可以路径回溯......有点傻...

原本是dijk1和dijk2(我说顺序),然后发现我改完上面那个之后莫名过了4个test(原本是俩),然后发现题读错了,在那改距离判定,然后还是wa,然后发现俩函数的判定反了...

要么距离优先要么时间优先不是嘛,那就主要次要变量换着存,之前打了仨结构体,然后发现不用这么多,嘤嘤嘤.

#include<bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
const int maxn = 503;
int n,m,s,en;
bool vis[maxn];
int dis[maxn];
int pre[maxn];
int cnt[maxn];
struct node
{
    int v,c,di;
    node()
    {
        v = c = di = 0;
    }
    node(int a,int b)
    {
        v = a;
        c = b;
        di = 0;
    }
    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];//dis
vector<node>vec2[maxn];//time
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));
}
void dijkstra2()
{
    for(int i=0; i<n; i++)
        pre[i] = i;
    memset(vis,false,sizeof(vis));
    for(int i=0; i<=n; i++)
    {
        dis[i] = INF;
        cnt[i] = 0;
    }
    while(!que.empty())
        que.pop();
    dis[s] = 0;
    pre[s] = -1;
    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 juli = vec2[u][i].di;
            if(!vis[v] && dis[v] > dis[u]+cost)
            {
                cnt[v] = cnt[u]+juli;
                dis[v] = dis[u] + cost;
                pre[v] = u;
                que.push(node(v,dis[v]));
            }
            else if(!vis[v] && dis[v] == dis[u]+cost)
            {
                if(cnt[v] > cnt[u] + juli)
                {
                    pre[v] = u;
                    cnt[v] = cnt[u]+juli;
                }
            }
        }
    }
}
void dijkstra1()
{
    for(int i=0; i<n; i++)
        pre[i] = i;
    memset(vis,false,sizeof(vis));
    for(int i=0; i<=n; i++)
        {
            dis[i] = INF;
            cnt[i] = INF;
        }
    while(!que.empty())
        que.pop();


    dis[s] = 0;
    pre[s] = -1;
    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)
            {
                cnt[v] = cnt[u]+1;
                dis[v] = dis[u] + cost;
                pre[v] = u;
                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;
                }
            }
        }
    }
}
bool panduan(int s1[],int s2[],int a,int b)
{
    if(a != b)
        return false;
    for(int i=0; i<a; i++)
    {
        if(s1[i] != s2[i])
            return false;
    }
    return true;
}
int main()
{
    //freopen("in.txt","r",stdin);
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    //freopen("in.txt","r",stdin);
    cin>>n>>m;
    int a,b,c,d,e;
    for(int i=0; i<m; i++)
    {
        cin>>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);
    }
//    for(int i=0;i<vec1[3].size();i++)
//        cout<<"is : "<<vec1[3][i].c<<" ";
    cin>>s>>en;
    dijkstra2();
    int ss1[maxn];
    int ss2[maxn];
    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(panduan(ss1,ss2,pos1,pos2))
    {
        cout<<"Time = "<<dis1<<"; ";
        cout<<"Distance = "<<dis[en]<<": ";
        cout<<s;
        for(int i=pos2-1; i>=0; i--)
            cout<<" => "<<ss2[i];
        cout<<endl;
        return 0;
    }
    cout<<"Time = "<<dis1<<": ";
    cout<<s;
    for(int i=pos1-1; i>=0; i--)
        cout<<" => "<<ss1[i];
    cout<<endl;
    cout<<"Distance = "<<dis[en]<<": ";
    cout<<s;
    for(int i=pos2-1; i>=0; i--)
        cout<<" => "<<ss2[i];
    cout<<endl;
    return 0;
}