题目:
本题要求你实现一个天梯赛专属在线地图,队员输入自己学校所在地和赛场地点后,该地图应该推荐两条路线:一条是最快到达路线;一条是最短距离的路线。题目保证对任意的查询请求,地图上都至少存在一条可达路线。
输入格式:
输入在第一行给出两个正整数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
思路:
两次Dijkstra求最短路,在求的过程中记录先节点
代码:
#include<bits/stdc++.h>
using namespace std;
int n,m;
int be,en;
//存距离和时间
int s[505][505],tim[505][505];
//是否加入了
int vis[505];
//先节点,总距离,总时间
int prel[505],pret[505],suml[505],sumt[505];
int anst[505],ansd[505];
const int inf=999999999;
//寻找最短时间的
void dijkstraT() {
vis[be]=1;
int dis[505];
for(int i=0; i<n; i++) {
dis[i]=inf;
dis[be]=0;
sumt[be]=0;
for(int i=0; i<n; i++) {
int misign=inf;
int signid=be;
for(int i=0; i<n; i++) {
if(sumt[i]<misign&&vis[i]==0) {
misign=sumt[i];
signid=i;
}
}
vis[signid]=1;
for(int i=0; i<n; i++) {
// 如果时间更短
if(sumt[i]>sumt[signid]+tim[signid][i]) {
pret[i]=signid;
sumt[i]=sumt[signid]+tim[signid][i];
dis[i]=dis[signid]+s[signid][i];
}
// 如果时间一样,但距离更短
else if(sumt[i]==sumt[signid]+tim[signid][i]&&dis[i]>dis[signid]+s[signid][i]) {
pret[i]=signid;
sumt[i]=sumt[signid]+tim[signid][i];
dis[i]=dis[signid]+s[signid][i];
}
}
}
}
}
//寻找距离最短
void dijkstraD(){
int node[505];
for(int i=0;i<n;i++){
node[i]=0;
}
memset(vis,0,sizeof(vis));
vis[be]=1;
suml[be]=0;
for(int i=0;i<n;i++){
int signid=be;
int misign=inf;
for(int i=0;i<n;i++){
if(!vis[i]&&suml[i]<misign){
misign=suml[i];
signid=i;
}
}
vis[signid]=1;
for(int i=0;i<n;i++){
// 如果距离更短
if(suml[i]>suml[signid]+s[signid][i]){
prel[i]=signid;
suml[i]=suml[signid]+s[signid][i];
node[i]=node[signid]+1;
}
// 如果距离一样,但节点更少
else if(suml[i]==suml[signid]+s[signid][i]&&node[i]>node[signid]+1){
prel[i]=signid;
suml[i]=suml[signid]+s[signid][i];
node[i]=node[signid]+1;
}
}
}
}
int main() {
cin>>n>>m;
// 初始化
for(int i=0; i<n; i++) {
prel[i]=-1;
pret[i]=-1;
suml[i]=inf;
sumt[i]=inf;
anst[i]=-1;ansd[i]=-1;
vis[i]=0;
for(int j=0; j<n; j++) {
if(i==j) {
s[i][j]=0;
tim[i][j]=0;
} else {
s[i][j]=inf;
tim[i][j]=inf;
}
}
}
// 输入并转化成领接矩阵
for(int i=1; i<=m; i++) {
int a,b,x,l,t;
cin>>a>>b>>x>>l>>t;
if(x==1) {
s[a][b]=l;
tim[a][b]=t;
} else {
s[a][b]=l;
tim[a][b]=t;
s[b][a]=l;
tim[b][a]=t;
}
}
cin>>be>>en;
dijkstraT();
dijkstraD();
// 判断两次节点是否一样
int pr=en,id1=0,id2=0,flag=0;
while(pr!=-1) {
anst[id1]=pr;
id1++;
pr=pret[pr];
}
pr=en;
while(pr!=-1) {
ansd[id2]=pr;
if(anst[id2]!=pr){
flag=-1;
}
id2++;
pr=prel[pr];
}
// 两次节点一样
if(id1==id2&&flag==0){
cout<<"Time = "<<sumt[en]<<"; Distance = "<<suml[en]<<": ";
for(int i=id2-1;i>=0;i--){
cout<<ansd[i];
if(i!=0){
cout<<" => ";
}
}
}
// 两次节点不一样
else{
cout<<"Time = "<<sumt[en]<<": ";
for(int i=id1-1;i>=0;i--){
cout<<anst[i];
if(i!=0){
cout<<" => ";
}
}
cout<<endl;
cout<<"Distance = "<<suml[en]<<": ";
for(int i=id2-1;i>=0;i--){
cout<<ansd[i];
if(i!=0){
cout<<" => ";
}
}
}
return 0;
}