NC26257 小雨坐地铁

题目链接

题目

题目描述

小雨所在的城市一共有 \(m\) 条地铁线，分别标号为 1 号线，2 号线，……，m 号线。整个城市一共有 \(n\) 个车站，编号为 \(1 \sim n\) 。其中坐 i 号线需要花费 \(a_i\) 的价格，每坐一站就需要多花费 \(b_i\) 的价格。i 号线有 \(c_i\) 个车站，而且这 \(c_i\) 个车站都已知，如果某一站有多条地铁线经过，则可以在这一站换乘到另一条地铁线，并且能多次换乘。现在小雨想从第 s 个车站坐地铁到第 t 个车站，地铁等待时间忽略不计，求最少花费的价格，若不能到达输出 -1 。（地铁是双向的，所以 s 可能大于 t）

输入描述

第一行输入四个正整数 \(n,m,s,t\)，分别表示车站个数，地铁线数，起点站和终点站。
第二行到第 \(m + 1\) 行，每行前三个数为 \(a_i,b_i,c_i\)，分别表示坐 i 号线的价格，i 号线每坐一站多花的价格，i 号线车站个数。接下来 \(c_i\) 个数，表示 i 号线的每一个车站的编号，单调递增。

输出描述

共一行，一个数表示最小花费，若不能到达输出 -1 。

示例1

输入

5 2 1 4
2 2 3 1 3 5
2 1 4 2 3 4 5

输出

7

说明

坐 1 号线：花费 2；

\(1 \rightarrow 3\) ：花费 2；

换乘 2 号线：花费 2；

\(3 \rightarrow 4\) ：花费 1；

所以最小总花费为 7 。

备注

\(1 \leq n \leq 10^3, 1 \leq m \leq 500,1 \leq s,t \leq n\)

\(1 \leq a_i,b_i \leq 100,1 \leq c_i \leq n,\sum\limits_{i = 1}^m c_i \leq 10^5\)

题解

知识点：图论建模，最短路。

利用分层图建模，每条地铁单独建立 \(n\) 个点。再额外建立 \(n\) 个点作为换乘的点，连接每条地铁每站。

最后跑最短路即可。

时间复杂度 \(O((nm+\sum c_i)\log\sum c_i)\)

空间复杂度 \(O(nm+\sum c_i)\)

代码

#include <bits/stdc++.h>
#define ll long long

using namespace std;

const int N = 1e3 * 501 + 7, M = 4 * 1e5 + 7;

template<class T>
struct Graph {
    struct edge {
        int v, nxt;
        T w;
    };
    int idx;
    vector<int> h;
    vector<edge> e;

    Graph(int n, int m) :idx(0), h(n + 1), e(m + 1) {}

    void clear(int n, int m) {//全局使用时清零，确定范围防止超时
        idx = 0;
        h.assign(n + 1, 0);
        e.assign(m + 1, { 0,0,0 });
    }

    void add(int u, int v, T w) {
        e[++idx] = edge{ v,h[u],w };
        h[u] = idx;
    }
};
Graph<int> g(N, M);

bool vis[N];
int dis[N];
struct node {
    int v, w;
    friend bool operator<(node a, node b) {
        return a.w > b.w;
    }
};
priority_queue<node> pq;
void dijkstra(int st) {
    memset(dis, 0x3f, sizeof(dis));
    dis[st] = 0;
    pq.push({ st,0 });
    while (!pq.empty()) {
        int u = pq.top().v;
        pq.pop();
        if (vis[u]) continue;
        vis[u] = 1;
        for (int i = g.h[u];i;i = g.e[i].nxt) {
            int v = g.e[i].v, w = g.e[i].w;
            if (dis[v] > dis[u] + w) {
                dis[v] = dis[u] + w;
                pq.push({ v,dis[v] });
            }
        }
    }
}

int main() {
    std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int n, m, s, t;
    cin >> n >> m >> s >> t;
    for (int i = 1;i <= m;i++) {
        int a, b, c;
        cin >> a >> b >> c;
        int pre = 0;
        while (c--) {
            int x;
            cin >> x;
            if (pre) {
                g.add(pre + i * n, x + i * n, b);
                g.add(x + i * n, pre + i * n, b);
            }
            //每层用以站点在不同线路连接，第一层当作站点，用以换乘
            g.add(x, x + i * n, a);
            g.add(x + i * n, x, 0);
            pre = x;
        }
    }
    dijkstra(s);
    cout << (dis[t] > 1e9 ? -1 : dis[t]) << '\n';

    return 0;
}