本题要求你实现一个天梯赛专属在线地图,队员输入自己学校所在地和赛场地点后,该地图应该推荐两条路线:一条是最快到达路线;一条是最短距离的路线。题目保证对任意的查询请求,地图上都至少存在一条可达路线。
输入格式:
输入在第一行给出两个正整数
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
网上没有发现邻接表写法,自己喜欢邻接表写,就发出来了。
核心思路 就是在 加入已经确定最短的的点的集合 时候顺便把最短的距离啊(用Min表示)和最少的点啊(用cnt表示)给顺带处理了,也借此保留了所经过路径(用pre表示)
#include<bits/stdc++.h> using namespace std; const int N = 3e5 +10; #define PII pair<int,int> #define fir first #define sec second int e[N],ne[N],idx,w[N],w1[N],h[N]; int dist[N],dist1[N]; int st[N],st1[N]; int pre[N],pre1[N]; int Min[N],cnt1[N]; int v[N]; int con[N],con1[N]; void add(int a, int b,int c,int d){ e[idx]=b; ne[idx] = h[a]; w[idx] = c; w1[idx] = d; h[a] = idx++; } void init() { memset(dist,0x3f,sizeof dist); memset(h,-1,sizeof h); memset(dist1,0x3f,sizeof dist1); memset(pre,-1,sizeof pre); memset(pre1,-1,sizeof pre1); memset(con,-1,sizeof con); memset(con1,-1,sizeof con1); } void dijkstra(int s){ priority_queue<PII,vector<PII>,greater<PII>>heap; heap.push({0,s}); dist[s] = 0; while(!heap.empty()){ auto t = heap.top(); heap.pop(); int dis = t.fir , ver = t.sec; if(st[ver])continue; st[ver] =1; for(int i = h[ver];~i;i = ne[i]){ int k = e[i]; if(dist[ver] + w[i] == dist[k]){//核心代码 if(Min[k] > Min[ver] + w1[i]){ Min[k] = Min[ver] + w1[i]; pre[k] = ver; } }//核心代码 if(dist[ver] + w[i] < dist[k]){ dist[k] = dist[ver] + w[i]; heap.push({dist[k],k}); Min[k] = Min[ver] + w1[i]; pre[k] =ver; } } } } void print1(int x){ if(pre1[x] == -1){ cout << x ; return; } print1(pre1[x]); cout <<" => "<< x; } void print(int x){ if(pre[x] == -1){ cout << x ; return; } print(pre[x]); cout <<" => "<< x; } void Dijkstra(int s){ priority_queue<PII,vector<PII>,greater<PII>>heap; heap.push({0,s}); dist1[s] = 0; while(!heap.empty()){ auto t = heap.top(); heap.pop(); int dis = t.fir , ver = t.sec; if(st1[ver])continue; st1[ver] =1; for(int i = h[ver];~i;i = ne[i]){ int k = e[i]; if(dist1[ver] + w1[i] == dist1[k]){//核心代码 if(cnt1[k] > cnt1[ver]+1){ pre1[k] = ver; } }//核心代码 if(dist1[ver] + w1[i] < dist1[k]){ dist1[k] = dist1[ver] + w1[i]; heap.push({dist1[k],k}); cnt1[k] =cnt1[ver]+1; pre1[k] = ver; } } } } int main() { int n,m;cin >> n >> m; init(); while(m--){ int a,b,c,d,e;cin >> a >> b >> c >> d >> e;//V1 V2 one-way d: length e: time if(c){ add(a,b,e,d); }else { add(a,b,e,d); add(b,a,e,d); } } int s , end;cin >> s >> end; dijkstra(s); Dijkstra(s); con[0] = end;con1[0] = end; int Idx=1,Idx1=1; for(int i = pre[end];~i;i=pre[i]){ con[Idx++] = i; } for(int i = pre1[end];~i;i=pre1[i]){ con1[Idx1++] = i; } int t =0; for(int i = 0;;i++){ if(con[i]==-1&&con1[i]==-1)t=1; if(con[i]!=con1[i])break; } if(t){ cout << "Time = "<< dist[end] << "; Distance = "<<dist1[end] << ": "; print(end); return 0; } cout << "Time = "<<dist[end]<<": "; print(end); cout <<endl; cout << "Distance = "<<dist1[end]<<": "; print1(end); }