PAT天梯赛练习题——L3-007. 天梯地图(多边权SPFA)

时间:2023-02-23 10:54:17

L3-007. 天梯地图

时间限制300 ms
内存限制65536 kB
代码长度限制8000 B
判题程序Standard作者陈越

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

输入格式:

输入在第一行给出两个正整数N(2 <= N <=500)和M,分别为地图中所有标记地点的个数和连接地点的道路条数。随后M行,每行按如下格式给出一条道路的信息:

V1 V2 one-way length time

其中V1V2是道路的两个端点的编号(从0到N-1);如果该道路是从V1V2的单行线,则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
一开始把两条路径的选择方法看成一样的了。以为时间相同时也选节点数最少的,中间一组一直答案错误,然后改完发现最后一组仍然是段错误估计是数组越界,然后不停地加边数,最后段错误太多次直接给他25W就过了。

代码:

#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<sstream>
#include<cstring>
#include<cstdio>
#include<string>
#include<deque>
#include<stack>
#include<cmath>
#include<queue>
#include<set>
#include<map>
using namespace std;
#define INF 0x3f3f3f3f
#define MM(x) memset(x,0,sizeof(x))
#define MMINF(x) memset(x,INF,sizeof(x))
typedef long long LL;
const double PI=acos(-1.0);
const int N=510,M=250010;
typedef pair<int,int> pii;

struct info
{
int to;
int len;
int time;
int pre;
}E[M];
int head[M*2],cnt;

void add(int s,int t,int dx,int costt)
{
E[cnt].to=t;
E[cnt].len=dx;
E[cnt].time=costt;
E[cnt].pre=head[s];
head[s]=cnt++;
}
struct road
{
int dx;
int step;
int pre;
}dlen[N];
struct costtime
{
int sumtime;
int dx;
int pre;
}dt[N];
void init(int n)
{
memset(head,-1,sizeof(head));
cnt=0;
for (int i=0; i<N; i++)
{
dlen[i].dx=INF;
dlen[i].step=1;
dlen[i].pre=-1;
dt[i].sumtime=INF;
dt[i].dx=INF;
dt[i].pre=-1;
}
}
void LEN_spfa(int s)
{
dlen[s].dx=0;
priority_queue<pii>Q;
Q.push(pii(-dlen[s].dx,s));
while (!Q.empty())
{
int now=Q.top().second;
Q.pop();
for (int i=head[now]; i!=-1; i=E[i].pre)
{
int v=E[i].to;
if((dlen[v].dx>dlen[now].dx+E[i].len)||
(dlen[v].dx==dlen[now].dx+E[i].len&&dlen[v].step>dlen[now].step+1))
{
dlen[v].dx=dlen[now].dx+E[i].len;
dlen[v].pre=now;
dlen[v].step=dlen[now].step+1;
Q.push(pii(-dlen[v].dx,v));
}
}
}
}
void TIME_spfa(int s)
{
dt[s].sumtime=0;
dt[s].dx=0;
priority_queue<pii>Q;
Q.push(pii(-dt[s].sumtime,s));
while (!Q.empty())
{
int now=Q.top().second;
Q.pop();
for (int i=head[now]; i!=-1; i=E[i].pre)
{
int v=E[i].to;
if((dt[v].sumtime>dt[now].sumtime+E[i].time)||
(dt[v].sumtime==dt[now].sumtime+E[i].time&&dt[v].dx>dt[now].dx+E[i].len))
{
dt[v].sumtime=dt[now].sumtime+E[i].time;
dt[v].pre=now;
dt[v].dx=dt[now].dx+E[i].len;
Q.push(pii(-dt[v].sumtime,v));
}
}
}
}
int main(void)
{
int v1,v2,ops,len,time,n,m,i,j,s,t;
while (~scanf("%d%d",&n,&m))
{
init(n);
for (i=0; i<m; i++)
{
scanf("%d%d%d%d%d",&v1,&v2,&ops,&len,&time);
add(v1,v2,len,time);
if(ops==0)
add(v2,v1,len,time);
}
scanf("%d%d",&s,&t);
LEN_spfa(s);
TIME_spfa(s);
vector<int>LEN_road,TIME_road;
for (i=t; i!=-1; i=dlen[i].pre)
{
LEN_road.push_back(i);
}
for (i=t; i!=-1; i=dt[i].pre)
{
TIME_road.push_back(i);
}
if(LEN_road==TIME_road)
{
printf("Time = %d; Distance = %d: ",dt[t].sumtime,dlen[t].dx);
for (i=LEN_road.size()-1; i>=0; i--)
{
if(i==0)
printf("%d\n",LEN_road[i]);
else
printf("%d => ",LEN_road[i]);
}
}
else
{
printf("Time = %d: ",dt[t].sumtime);
for (i=TIME_road.size()-1; i>=0; i--)
{
if(i==0)
printf("%d\n",TIME_road[i]);
else
printf("%d => ",TIME_road[i]);
}
printf("Distance = %d: ",dlen[t].dx);
for (i=LEN_road.size()-1; i>=0; i--)
{
if(i==0)
printf("%d\n",LEN_road[i]);
else
printf("%d => ",LEN_road[i]);
}
}
}
return 0;
}