1131 Subway Map

题目描述
知识点： 堆优化版迪杰斯特拉
思路： 本题首先是建图问题。对于一条线路上的不同站点权值为经过的站点个数（这个思路很重要），然后这样子建图，使用邻接表建图。
之后，问题就转化为多关键字排序，找到权值最小的情况下，走过的边数最少的最短路。一条边就代表了一条地铁线路。
之后，由于边达到了一百万级别，点数为1万，无法使用邻接矩阵进行存储。
另外，点数为1万，传统迪杰斯特拉会超时。需要使用堆优化。
具体就是找最短dist的时候使用堆来维护，每次弹出最小的没被访问过的dist，如果被访问过就continue。

#include<iostream>
#include<queue>
#include<cstring>
#include<vector>
#define x first
#define y second
#define PIL pair<int,int>
using namespace std;
const int M = 2e6+10,N = 11000;
int h[N],e[M],ne[M],line[M],w[M],idx,stops[N];
int dist[N],pre[N],st[N],cnt[N];
string string_path[N];
string get_num(int c){
    string s = to_string(c);
    while(s.size() < 4) s = '0' + s;
    return s;
}
void add(int a,int b,int c,int d){
    ne[idx] = h[a],line[idx] = d,w[idx] = c,e[idx] = b,h[a] = idx++;
}
void dijkstra(int s,int end){
    memset(dist,0x3f,sizeof dist);
    memset(st,0,sizeof st);
    memset(cnt,0,sizeof cnt);
    memset(pre,-1,sizeof pre);
    priority_queue<PIL,vector<PIL>,greater<PIL>> heap;
    dist[s] = cnt[s] = 0;
    heap.push({0,s});
    while(heap.size() > 0){
        auto top = heap.top();
        heap.pop();
        int ver = top.y;
        if(ver == end) break;
        if(st[ver]) continue;
        st[ver] = true;
        for(int i = h[ver];i != -1;i = ne[i]){
            int u = e[i];
            if(dist[u] > dist[ver] + w[i]){//只要一换边，一定换乘
                dist[u] = dist[ver] + w[i];
                pre[u] = ver;
                cnt[u] = cnt[ver] + 1;
                string_path[u] = "Take Line#" + to_string(line[i])
                    + " from "+ get_num(ver) +" to " + get_num(u) + ".";
                heap.push({dist[u],u});
            }else if(dist[u] == dist[ver] + w[i] && cnt[u] > cnt[ver] + 1){
                cnt[u] = cnt[ver] + 1;
                pre[u] = ver;
                string_path[u] = "Take Line#" + to_string(line[i])
                    + " from "+ get_num(ver) +" to " + get_num(u) + ".";                
            }
        }
    }
    printf("%d\n",dist[end]);
    vector<string> path;
    while(end != s){
        path.push_back(string_path[end]);
        end = pre[end];
    }
    for(int i = path.size()-1;i >= 0;i--) printf("%s\n",path[i].c_str());
}
int main(){
    int n;
    scanf("%d",&n);
    memset(h,-1,sizeof h);
    for(int k = 1;k <= n;k++){
        int m;
        scanf("%d",&m);
        for(int i = 0;i < m;i++) scanf("%d",&stops[i]);
        //建图
        for(int i = 0;i < m;i++){
            for(int j = i+1;j < m;j++){
                int length = j - i;
                if(stops[0] == stops[m-1]) length = min(length,m - j + i - 1);
                add(stops[i],stops[j],length,k);
                add(stops[j],stops[i],length,k);  
            }
        }
    }
    int q;
    scanf("%d",&q);
    while(q--){
        int s,e;
        scanf("%d%d",&s,&e);
        dijkstra(s,e);
    }
    return 0;
}