50: Luogu P4568 分层图

分层图最短路模板

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <ctime>
#include <queue>
#include <cstring>

using namespace std;

const int M = 2e6 + 5e5 + ;

#define gc getchar()
inline int read() {
    int x = , f = ; char c = gc;
    while(c < '' || c > '') {if(c == '-') f = -; c = gc;}
    while(c >= '' && c <= '') x = x *  + c - '', c = gc;
    return x * f;
}

int dis[M], head[M], cnt, vis[M];
struct Node {
    int u, v, nxt, w;
} G[M];

struct Node1 {
    int u, dis;
    bool operator < (const Node1 a) const {
        return this->dis > a.dis;
    }
};

priority_queue <Node1> Q;

int n, m, k;
int s, t;

void Link(int u, int v, int w) {
    G[++ cnt].v = v; G[cnt].u = u; G[cnt].w = w; G[cnt].nxt = head[u]; head[u] = cnt;
}

void Dijkstra() {
    memset(dis, 0x3f, sizeof dis);
    Q.push((Node1){s, });
    dis[s] = ;
    while(!Q.empty()) {
        Node1 tp = Q.top();
        Q.pop();
        if(vis[tp.u]) continue;
        vis[tp.u] = ;
        for(int i = head[tp.u]; ~ i; i = G[i].nxt) {
            int v = G[i].v;
            if(dis[v] > dis[tp.u] + G[i].w) {
                dis[v] = dis[tp.u] + G[i].w;
                 Q.push((Node1) {v, dis[v]});
            }
        }
    }
}

int main() {
    n = read(), m = read(), k = read();
    for(int i = ; i <= n * k + n; i ++) head[i] = -;
    s = read() + , t = read() + ;
    for(int i = ; i <= m; i ++) {
        int u = read() + , v = read() + , w = read();
        for(int j = ; j < k; j ++) {
            Link(j * n + u, j * n + v, w);
            Link(j * n + v, j * n + u, w);
            Link(j * n + u, (j + ) * n + v, );
            Link(j * n + v, (j + ) * n + u, );
        }
        Link(n * k + u, n * k + v, w);
        Link(n * k + v, n * k + u, w);
    }
    Dijkstra();
    int Ans = ( << );
    for(int i = ; i <= k; i ++) {
        Ans = min(Ans, dis[i * n + t]);
    }
    cout << Ans;
    return ;
}