PAT甲题题解-1111. Online Map (30)-PAT甲级真题(模板题,两次Dijkstra,同时记下最短路径)

时间:2022-02-22 10:51:20

题意:给了图,以及s和t,让你求s到t花费的最短路程、最短时间,以及输出对应的路径。

     对于最短路程,如果路程一样,输出时间最少的。

   对于最短时间,如果时间一样,输出节点数最少的。

     如果最短路程和最短时间路径一样,合并输出一次即可。

纯粹就是练习dijkstra,没什么难的。

第一次dijkstra求最短路程,记录下每个节点的路程和时间。

第二次dijkstra求最短时间,记录下每个节点的时间和经过的节点数。

pre数组用来存储前驱节点,保存路径

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string.h>
#include <queue>
#define INF 0x3f3f3f3f
using namespace std; const int maxn=;
int n,m;
int dis[maxn];
int distime[maxn];
int nums[maxn];
int pre[maxn];
int vis[maxn];
struct Edge{
int w,time;
}edge[maxn][maxn];
struct disNode{
int dis; //统计距离
int time; //统计时间
int u;
bool operator<(const disNode tmp)const{
if(dis==tmp.dis)
return time>tmp.time;
else
return dis>tmp.dis;
}
}; struct timeNode{
int time; //统计时间
int nums; //统计路径的节点数
int u;
bool operator<(const timeNode tmp)const{
if(time==tmp.time)
return nums>tmp.nums;
else
return time>tmp.time;
}
};
/*
求取最短路径,如果最短距离一样,取时间花费最小的
*/
void dijkstraDis(int s){
int u;
for(int i=;i<n;i++){
dis[i]=INF;
vis[i]=;
pre[i]=-;
}
priority_queue<disNode>q;
disNode node;
node.u=s;
node.dis=dis[s]=;
node.time=dis[s]=;
q.push(node);
while(!q.empty()){
node=q.top();
q.pop();
u=node.u;
vis[u]=;
for(int v=;v<n;v++){
if(!vis[v] && edge[u][v].w!=-){
//一开始都只判断了前一个,没判断后面一个。。。导致有个样例没过
if((dis[u]+edge[u][v].w<dis[v])||(dis[u]+edge[u][v].w==dis[v] && distime[u]+edge[u][v].time<distime[v])){
dis[v]=dis[u]+edge[u][v].w;
distime[v]=distime[u]+edge[u][v].time;
node.dis=dis[v];
node.time=distime[v];
node.u=v;
pre[v]=u;
q.push(node);
}
}
}
}
}
/*
求取时间最小的,如果时间一样,取路径节点数最小的。
*/
void dijkstraTime(int s){
int u;
for(int i=;i<n;i++){
distime[i]=INF;
vis[i]=;
pre[i]=-;
nums[i]=;
}
priority_queue<timeNode>q;
timeNode node;
node.u=s;
node.time=distime[s]=;
node.nums=nums[s]=;
q.push(node);
while(!q.empty()){
node=q.top();
q.pop();
u=node.u;
vis[u]=;
for(int v=;v<n;v++){
if(!vis[v] && edge[u][v].w!=-){
if((distime[u]+edge[u][v].time<distime[v])||(distime[u]+edge[u][v].time==distime[v] && nums[u]+<nums[v])){
distime[v]=distime[u]+edge[u][v].time;
nums[v]=nums[u]+;
node.time=distime[v];
node.nums=nums[v];
node.u=v;
pre[v]=u;
q.push(node);
}
}
}
}
} int main()
{
int v1,v2,a,b,c;
int s,t;
scanf("%d %d",&n,&m);
for(int i=;i<n;i++){
for(int j=;j<n;j++){
edge[i][j].w=-;
}
}
for(int i=;i<m;i++){
scanf("%d %d %d %d %d",&v1,&v2,&a,&b,&c);
edge[v1][v2].w=b;
edge[v1][v2].time=c;
if(a==){
edge[v2][v1].w=b;
edge[v2][v1].time=c;
}
}
scanf("%d %d",&s,&t);
dijkstraDis(s);
int disAns=dis[t];
int disPath[n+];
int disCnt=;
//最短路径
int u=t;
disPath[disCnt]=u;
disCnt++;
while(pre[u]!=-){
disPath[disCnt]=pre[u];
u=pre[u];
disCnt++;
} dijkstraTime(s);
int timeAns=distime[t];
int timePath[n+];
int timeCnt=;
//最短时间路径
u=t;
timePath[timeCnt]=u;
timeCnt++;
while(pre[u]!=-){
timePath[timeCnt]=pre[u];
u=pre[u];
timeCnt++;
} bool isSame=true;
if(disCnt==timeCnt){
for(int i=;i<disCnt;i++){
if(disPath[i]!=timePath[i]){
isSame=false;
break;
}
}
}
else
isSame=false; if(isSame){
printf("Distance = %d; Time = %d: %d",disAns,timeAns,s);
for(int i=disCnt-;i>=;i--){
printf(" -> %d",disPath[i]);
}
}
else{
printf("Distance = %d: %d",disAns,s);
for(int i=disCnt-;i>=;i--){
printf(" -> %d",disPath[i]);
}
printf("\n");
printf("Time = %d: %d",timeAns,s);
for(int i=timeCnt-;i>=;i--){
printf(" -> %d",timePath[i]);
}
} return ;
}