L3-007. 天梯地图

L3-007. 天梯地图

时间限制

300 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

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

输入格式：

输入在第一行给出两个正整数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: 起点 => 节点₁ => ... => 终点

然后在下一行按下列格式输出最短距离D和用节点编号表示的路线：

Distance = D: 起点 => 节点₁ => ... => 终点

如果最快到达路线不唯一，则输出几条最快路线中最短的那条，题目保证这条路线是唯一的。而如果最短距离的路线不唯一，则输出途径节点数最少的那条，题目保证这条路线是唯一的。

如果这两条路线是完全一样的，则按下列格式输出：

Time = T; Distance = D: 起点 => 节点₁ => ... => 终点

输入样例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

---

提交代码

分析：和之前那个紧急救援同样的做法，用一个附带数组和d数组同步变化，注意的是节点数和道路数不一样，用=而不是+=，因为节点数不是增加的，而是唯一的，道路数那个题求的是一共几条路,所以两个题不太一样。

坑点：临界数组初始化用N*N，别用n。两个djk或者两个输出的时候变量名一定要修改!!!!!

#include <bits/stdc++.h>

using namespace std;
int n,m;
const int N=505;
const int INF=99999999;
int mp[N][N];//长度
int mp2[N][N];//时间
int d1[N],d11[N],d2[N],d21[N];
vector<int> ans1;
vector<int> ans2;
int pre[N],pre2[N];
int visit[N];
void djk(int start)
{
    fill(d2,d2+n,INF);
    memset(d21,0,sizeof(d21));
    memset(visit,0,sizeof(visit));
    d2[start]=0;
    d21[start]=0;

    for(int i=0; i<n; i++)
    {
        int u=-1;
        int MIN=INF;
        for(int j=0; j<n; j++)
        {
            if(d2[j]<MIN &&!visit[j])
            {
                u=j;
                MIN=d2[j];
            }
        }

        if(u==-1) return;
        visit[u]=1;
        for(int j=0; j<n; j++)
        {
            if(!visit[j]&&d2[u]+mp2[u][j]<d2[j])
            {

                d2[j]= d2[u]+mp2[u][j];
                d21[j]=d21[u]+mp[u][j];
                pre[j]=u;
            }
            else if(!visit[j]&&d2[u]+mp2[u][j]==d2[j])
            {
                if(d21[j]>d21[u]+mp[u][j])
                {
                    d21[j]=d21[u]+mp[u][j];
                    pre[j]=u;
                }
            }
        }
    }

}
void djk2(int start)
{
    fill(d1,d1+n,INF);
    memset(d11,0,sizeof(d11));
    memset(visit,0,sizeof(visit));
    d1[start]=0;
    d11[start]=1;

    for(int i=0; i<n; i++)
    {
        int u=-1;
        int MIN=INF;
        for(int j=0; j<n; j++)
        {
            if(d1[j]<MIN &&!visit[j])
            {
                u=j;
                MIN=d1[j];
            }
        }
        if(u==-1) return;
        visit[u]=1;
        for(int j=0; j<n; j++)
        {
            if(!visit[j]&&d1[u]+mp[u][j]<d1[j])
            {
                d1[j]= d1[u]+mp[u][j];
                d11[j]=d11[u]+1;
                pre2[j]=u;
            }
            else if(!visit[j]&&d1[u]+mp[u][j]==d1[j])
            {
                d1[j]= d1[u]+mp[u][j];
                if(d11[j]>d11[u]+1)
                {
                    d11[j]=d11[u]+1;
                    pre2[j]=u;
                }
            }
        }
    }

}
void print(int start,int target)
{
    if(start==target)
    {
        ans1.push_back(start);
        return;
    }
    print(start,pre[target]);
    ans1.push_back(target);
}
void print2(int start,int target)
{
    if(start==target)
    {
        ans2.push_back(start);
        return;
    }
    print2(start,pre2[target]);
    ans2.push_back(target);
}
int main()
{
    cin>>n>>m;
    fill(mp[0],mp[0]+N*N,INF);
    fill(mp2[0],mp2[0]+N*N,INF);
    int v1,v2,jud,l,t;
    for(int i=0; i<m; i++)
    {
        scanf("%d%d%d%d%d",&v1,&v2,&jud,&l,&t);
        if(jud)
        {
            mp[v1][v2]=l;
            mp2[v1][v2]=t;
        }
        else
        {
            mp[v1][v2]=l;
            mp[v2][v1]=l;
            mp2[v1][v2]=t;
            mp2[v2][v1]=t;
        }
    }
    int start,target;
    cin>>start>>target;
    djk(start);
    djk2(start);
    print(start,target);
    print2(start,target);

    if(ans1==ans2)
    {
        printf("Time = %d; Distance = %d: ",d2[target],d1[target]);
        for(int i=0; i<ans1.size(); i++)
        {
            cout<<ans1[i];

            if(i!=ans1.size()-1) cout<<" => ";
        }
    }
    else
    {
        printf("Time = %d: ",d2[target]);
        for(int i=0; i<ans1.size(); i++)
        {
            cout<<ans1[i];

            if(i!=ans1.size()-1) cout<<" => ";
        }
        cout<<endl;
        printf("Distance = %d: ",d1[target]);
        for(int i=0; i<ans2.size(); i++)
        {
            cout<<ans2[i];
            if(i!=ans2.size()-1) cout<<" => ";
        }

    }

    return 0;
}