【ACWing】176. 装满的油箱

题目地址：

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

有$N$个城市（编号$0,1,\dots ,N-1$）和$M$条道路，构成一张无向图。在每个城市里边都有一个加油站，不同的加油站的单位油价不一样。现在你需要回答不超过$100$个问题，在每个问题中，请计算出一架油箱容量为$C$的车子，从起点城市$S$开到终点城市$E$至少要花多少油钱？注意：假定车子初始时油箱是空的。

输入格式：
第一行包含两个整数$N$和$M$。第二行包含$N$个整数，代表$N$个城市的单位油价，第$i$个数即为第$i$个城市的油价$p_i$。接下来$M$行，每行包括三个整数$u,v,d$，表示城市$u$与城市$v$之间存在道路，且车子从$u$到$v$需要消耗的油量为$d$。接下来一行包含一个整数$q$，代表问题数量。接下来$q$行，每行包含三个整数$C$、$S$、$E$，分别表示车子油箱容量$C$、起点城市$S$、终点城市$E$。

输出格式：
对于每个问题，输出一个整数，表示所需的最少油钱。如果无法从起点城市开到终点城市，则输出impossible。每个结果占一行。

数据范围：
$1\le N\le 1000$
$1\le M\le 10000$
$1\le p_i\le 100$
$1\le d\le 100$
$1\le C\le 100$

本质上是分层图最短路问题。考虑新图，其每个点是一个数对$(x,c)$，$x$代表城市编号，$c$代表在$x$时的油量。初始点是$(S,0)$，终点是$(E,c^{\prime })$。对于$(x,c)$，其到$(x,c+1),c+1\le C$有条边，边权为$p_x$，代表加一单位油需要的花费；其到$(y,c-d_{x\to y}),c\ge d_{x\to y},x\to y$有条边，边权为$0$，代表花费$d_{x\to y}$这么多油从$x$开到$y$。那么原图的每一个方案在新图中都有一个方案与之一一对应，从而新图里的最短路对应的方案就是原图中花费最少的方案，所以问题转化为在新图里求最短路，可以用Dijkstra算法来做。由于是稀疏图，可以用堆优化版本。代码如下：

#include <iostream>
#include <cstring>
#include <queue>
using namespace std;

struct Ver {
	// d是总花费，u是城市编号，c是当前车的油量
    int d, u, c;
    bool operator<(const Ver &v) const {
        return d > v.d;
    }
};
// 无向图，边要开两倍
const int N = 1010, M = 20010, C = 1010;
int n, m;
int h[N], e[M], ne[M], w[M], idx;
int price[N];
int dist[N][C];
bool vis[N][C];

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

int dijkstra(int c, int start, int end) {
    memset(dist, 0x3f, sizeof dist);
    memset(vis, 0, sizeof vis);
    
    priority_queue<Ver> heap;
    heap.push({0, start, 0});
    while (heap.size()) {
        auto t = heap.top(); heap.pop();
        if (t.u == end) return t.d;
        if (vis[t.u][t.c]) continue;
        vis[t.u][t.c] = true;
        // 如果能加油的话，可以走到(t.u, t.c + 1)这个点
        if (t.c < c) {
            if (dist[t.u][t.c + 1] > t.d + price[t.u]) {
                dist[t.u][t.c + 1] = t.d + price[t.u];
                heap.push({dist[t.u][t.c + 1], t.u, t.c + 1});
            }
        } 
        
        for (int i = h[t.u]; ~i; i = ne[i]) {
            int j = e[i];
            // 如果油量够的话，可以走到(j, t.c - w[i])这个点
            if (t.c >= w[i]) {
                if (dist[j][t.c - w[i]] > t.d) {
                    dist[j][t.c - w[i]] = t.d;
                    heap.push({dist[j][t.c - w[i]], j, t.c - w[i]});
                }
            }
        }
    }

    return -1;
}

int main() {
    scanf("%d%d", &n, &m);
    for (int i = 0; i < n; i++) scanf("%d", &price[i]);
    memset(h, -1, sizeof h);
    while (m--) {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        add(a, b, c), add(b, a, c);
    }

    int query;
    scanf("%d", &query);
    while (query--) {
        int c, s, e;
        scanf("%d%d%d", &c, &s, &e);
        int t = dijkstra(c, s, e);
        if (~t) printf("%d\n", t);
        else puts("impossible");
    }

    return 0;
}

每次询问时间复杂度$O(M\log (NC))$，空间$O(NC)$。