本题要求你实现一个天梯赛专属在线地图,队员输入自己学校所在地和赛场地点后,该地图应该推荐两条路线:一条是最快到达路线;一条是最短距离的路线。题目保证对任意的查询请求,地图上都至少存在一条可达路线。
输入格式:
输入在第一行给出两个正整数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
本题我采用两次bfs算法,第一次搜索最短距离,第二次搜索最短时间,并且用path数组存储路径,但不知道为何第五个样例点过不去,有没有眼细的小伙伴帮忙看看第五个样例点是啥问题。
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#include<map>
using namespace std;
const int N=510;
int e[N*N],en[N*N],h[N];
int dis[N*N],t[N*N];
int cnt=0;
int path1[N];
int path2[N];
int num1=0;
int num2=0;
int dd[N][N],tt[N][N];
int sum_d=0,sum_t=0;
void add(int i,int j,int length,int time)
{
e[cnt]=j;en[cnt]=h[i];h[i]=cnt;dis[cnt]=length;t[cnt]=time;cnt++;
}
void bfs_d(int i,int j)
{
int vis[N];
memset(vis,0x3f,sizeof vis);
vis[i]=0;
queue<int>q;
q.push(i);
path1[i]=i;
while(!q.empty())
{
int m=q.front();
q.pop();
for(int a=h[m];a!=-1;a=en[a])
{
int t=e[a];
if(vis[t]>vis[m]+dis[a])
{
vis[t]=vis[m]+dis[a];
path1[t]=m;
q.push(t);
}
else if(vis[t]==vis[m]+dis[m])//判断如果距离相等,选择结点少的
{
int x=0;
int y=0;
int obj1=t,obj2=m;
while(path1[obj1]!=i)
{
x++;
obj1=path1[obj1];
}
while(path1[obj2]!=i)
{
y++;
obj2=path1[obj2];
}
if(x>y+1)
{
path1[t]=m;
q.push(t);
}
}
}
}
sum_d=vis[j];//直接将最短的值传到sum变量中
}
void bfs_l(int i,int j)
{
int vis[N],d[N];
memset(vis,0x3f,sizeof vis);
memset(d,0x3f,sizeof d);
vis[i]=0;
d[i]=0;
queue<int>q;
q.push(i);
path2[i]=i;
while(!q.empty())
{
int m=q.front();
q.pop();
for(int a=h[m];a!=-1;a=en[a])
{
int s=e[a];
if(vis[s]>vis[m]+t[a])
{
vis[s]=vis[m]+t[a];
d[s]=d[m]+dis[a];//存储相应的距离
path2[s]=m;
q.push(s);
}
else if(vis[s]==vis[m]+t[a])//判断如果时间相等,选择距离少的
{
if(d[s]>d[m]+dis[a])
{
d[s]=d[m]+dis[a];
path2[s]=m;
q.push(s);
}
}
}
}
sum_t=vis[j];
}
int main()
{
int n,m;
cin>>n>>m;
memset(h,-1,sizeof h);
for(int i=0;i<m;i++)
{
int a,b,sign,length,time;
cin>>a>>b>>sign>>length>>time;
if(sign)
{
add(a,b,length,time);
dd[a][b]=length;
tt[a][b]=time;
}
else {
add(a,b,length,time);
add(b,a,length,time);
dd[a][b]=length;
tt[a][b]=time;
dd[b][a]=length;
tt[b][a]=time;
}
}
int a,b;
cin>>a>>b;
bfs_d(a,b);
bfs_l(a,b);
int res[N],res1[N];
int t=path1[b];
int num1=0,num2=0;
res[num1++]=b;
res1[num2++]=b;
while(t!=a)
{
res[num1++]=t;
t=path1[t];
}
res[num1++]=a;
t=path2[b];
while(t!=a)
{
res1[num2++]=t;
t=path2[t];
}
res1[num2++]=a;
int flag=1;
if(num1==num2) {
for (int i = 0; i < num1; i++) {
if (res[i] != res1[i]) {
flag = 0;
break;
}
}
if (!flag)
{
cout<<"Time = "<<sum_t<<": ";
for(int i=num2-1;i>=0;i--)
{
cout<<res1[i];
if(i!=0)
cout<<" => ";
}
cout<<endl;
cout<<"Distance = "<<sum_d<<": ";
for(int i=num1-1;i>=0;i--)
{
cout<<res[i];
if(i!=0)
cout<<" => ";
}
}
else
{
cout<<"Time = "<<sum_t<<"; Distance = "<<sum_d<<": ";
for(int i=num1-1;i>=0;i--)
{
cout<<res[i];
if(i!=0)
cout<<" => ";
}
}
}
else {
cout << "Time = " << sum_t << ": ";
for (int i = num2 - 1; i >= 0; i--) {
cout << res1[i];
if (i != 0)
cout << " => ";
}
cout<<endl;
cout << "Distance = " << sum_d << ": ";
for (int i = num1 - 1; i >= 0; i--) {
cout << res[i];
if (i != 0)
cout << " => ";
}
}
return 0;
}