PAT甲级1131 Subway Map (30分)：[C++题解]堆优化dijkstra、单源最短路、地铁地图、巧妙地建图套dijkstra模板！！

题目分析

原题：




来源：acwing

分析：

建图：所有能走到的点之间建立一条边，比如下面一条地铁线路有4站，它们是相通的，两两之间建一条边，边权是经过的站点数。

下面考虑一下使用哪种最短路算法。

每条线路最多100个点，如上这样建图，最多的边数为：$C_{100}^2<5000<10000$条边；最多有100条路线，最多10000个点，最多建立 10000 × 100 = 1e6条边，即100万条边。

使用朴素的dijkstra算法会超时$O(n^2)=10000^2=10^8超时$， 需要使用堆优化版本的dijkstra()算法$O(m\times logn)=1e6\times log10000=4e6不会超时$

限制1s的话，运算1e8次差不多；现在限制是0.4s，差不多运算 5e7次才不超时。

看题意：第一点，首先输出最少需要停靠的站点数量，这就是在我们的建图中最短路（每条边有权重，权重等于结点之间的站点数）；第二点，如果最快线路不唯一，则输出换乘次数最少的线路，换乘次数最少，在我们的图中等价于边数最少，因为同一条路线上的点只有一条边，边数越多，表示经过的线路越多。

综上：最短路取决于边权大小；换乘少取决于边数少。
建图如下：下图有两条地铁线路，分别有4个站点和5个站点，并且有1个换乘站，所以下面的几个点都是连通的，可以通过换乘到达。我们的建图原则是：在一条线路上的任意两站之间建一条边，边权为两者之间的站点数量。

堆优化版的dijkstra算法模板请移步最短路[Dijkstra和堆优化的Dijkstra][Bellman-Ford和SPFA][Floyd最短路]（更新中）

ac代码

#include<bits/stdc++.h>
using namespace std;
const int N =10010 , M =1e6+10;
typedef pair<int,int> PII;
# define x first
#define y second

int n , h[N],e[M],ne[M],w[M],idx;
int line[M];//每条边属于几号线
int stops[N];//站点
int dist[N];//
int cnt[N]; //记录经过的边数；换成少===经过的边数少
bool st[N];
int pre[N];
string info[N];//记录地铁线路信息


/*
邻接表建边：建立a到b的边
a和b是站点
c是权值，表示中间隔了几站
id是线路号line
*/

void add(int a ,int b, int c, int id){
    e[idx] =b,w[idx] =c,line[idx] =id, ne[idx] = h[a] ,h[a] =idx ++;
}

//4位站牌号，不足补上
string get_number(int x){
    string res  = to_string(x);
    while(res.size() <4) res = '0' + res;
    return res;
    
}

void dijkstra(int start, int end){
    
    memset(dist, 0x3f, sizeof dist);
    memset(cnt, 0 ,sizeof cnt);
    memset(st, 0 ,sizeof st);
    
    dist[start] = 0 ,cnt[start] =0;
    priority_queue<PII, vector<PII> ,greater<PII>> heap;//小根堆
    heap.push({0,start}); //加入起点{距离，编号}
    
    while(heap.size()){
        auto t = heap.top();
        heap.pop();
        
        int ver =t.y; //节点编号
        if(ver == end) break;
        if(st[ver]) continue;
        
        st[ver] =true;
        
        for(int i=h[ver]; i!= -1; i = ne[i]){
            int j  = e[i];
            //距离更短，更新
            if(dist[j] > dist[ver] + w[i]){
                dist[j] =dist[ver] +w[i];
                cnt[j] =cnt[ver] +1; //走过的边数
                pre[j] = ver;
                info[j] ="Take Line#"+to_string(line[i])+ 
                        " from "+ get_number(ver) +" to "+ get_number(j) +".";
                heap.push({dist[j] ,j});
            }
            //距离相同，看看
            else if( dist[j] == dist[ver] +w[i]){
                if(cnt[j] > cnt[ver] + 1){
                    cnt[j] =cnt[ver] +1;
                    pre[j] = ver;
                    info[j] ="Take Line#"+to_string(line[i])+ 
                            " from "+ get_number(ver) +" to "+ get_number(j) +".";
                    heap.push({dist[j],j});
                    
                }
            }
        }
    }
    
    cout<< dist[end]<<endl;
    vector<string > path;
    //倒序存入path
    for(int i = end; i != start; i =pre[i]) path.push_back(info[i]);
    
    for(int i =path.size()-1; i>=0 ;i--) cout<< path[i ] <<endl;
}

int main()
{
    cin >> n;
    memset(h ,-1 ,sizeof h); //邻接表初始化
    
    for(int i =1; i<= n; i++){ //n条线路
        int m;//站点
        cin >> m;
        for(int i = 0; i< m; i++) cin >> stops[i];
        //同一条线路上的站点，因为都是连通的，两两之间建边
        for(int j = 0; j< m; j++){
            for(int k =0; k< j; k++){
                int length;
                //如果不是环路
                if(stops[0] !=stops[m-1]) length = j - k;
                //如果是环路，取两边短的
                else length = min(j - k , m-1 - j +k);
                //建边的时候把长度也建好了，等于之间走了几站：length
                add(stops[j],stops[k],length, i);  
                add(stops[k],stops[j], length ,i);
            }
        }
        
    }
    
    int m;
     cin >> m;
     while(m--){
         int start ,end;
         cin >> start >> end;
         dijkstra(start, end);
     }
    
} 

题目链接

PAT甲级1131 Subway Map (30分)

https://www.acwing.com/problem/content/1626/