P4009 汽车加油行驶问题（分层图最短路）

最新推荐文章于 2021-08-06 09:49:15 发布

原创最新推荐文章于 2021-08-06 09:49:15 发布 · 705 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

最短路专栏收录该内容

29 篇文章

订阅专栏

本文介绍了一种解决汽车从网格起点到终点最小费用路径的问题算法。汽车需遵守特定规则，如仅沿网格边行驶，遇到油库必须加油等。算法通过建立多层图，将不同油量状态作为不同层，实现费用最小化。

题目描述
给定一个 N×N 的方形网格，设其左上角为起点◎，坐标(1,1)，X 轴向右为正， Y 轴向下为正，每个方格边长为 1 ，如图所示。

一辆汽车从起点◎出发驶向右下角终点▲，其坐标为 (N,N)。

在若干个网格交叉点处，设置了油库，可供汽车在行驶途中加油。汽车在行驶过程中应遵守如下规则:

汽车只能沿网格边行驶，装满油后能行驶 K 条网格边。出发时汽车已装满油，在起点与终点处不设油库。

汽车经过一条网格边时，若其 X 坐标或 Y 坐标减小，则应付费用 B ，否则免付费用。

汽车在行驶过程中遇油库则应加满油并付加油费用 A。

在需要时可在网格点处增设油库，并付增设油库费用 C(不含加油费用A )。

设计一个算法，求出汽车从起点出发到达终点所付的最小费用。
思路：
可以分为k+1层图，第一层图表示满油，接下来每层图油-1，建图。
当此处没加油站，把第i层连向i+1层下一个位置。同时向第0层连一个权值a+c的边表示修建加油站加油。
若有加油站，则要注意，题目里说了是强制消费，所以必须加油，怎么保证必须加油呢，建完从i层到0层的权值为a的边后，只建从0到1的下一个位置的边，这样他只能从第i层下来走0-1的边，不然就不连通了。

#include<bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
#define MP make_pair
#define PII pair<int, int>
using namespace std;
const int N = 1e6 + 10;
int n, k, a, b, c, h[N], cnt, s, vis[N], dis[N];
struct point {
    int v, w, nt;
} no[N];
void add(int u, int v, int w) {
    no[cnt] = point{v, w, h[u]};
    h[u] = cnt++;
}
int co(int x, int y, int k) {
    return n * n * k + ((x - 1) * n) + y;
}
void dijkstra() {
    memset(dis, 0x3f, sizeof dis);
    priority_queue<PII, vector<PII>, greater<PII> > q;
    q.push(MP(0, s));
    dis[s] = 0;
    while(!q.empty()) {
        PII e = q.top();
        q.pop();
        int u = e.second, val = e.first;
        if(vis[u])
            continue;
        vis[u] = 1;
        for(int i = h[u]; ~i; i = no[i].nt) {
            int v = no[i].v;
            if(!vis[v] && dis[v] > dis[u] + no[i].w) {
                dis[v] = dis[u] + no[i].w;
                q.push(MP(dis[v], v));
            }
        }
    }
}
int main() {
    memset(h, -1, sizeof h);
    scanf("%d%d%d%d%d", &n, &k, &a, &b, &c);
    s = 1;
    for(int f, i = 1; i <= n; i++)
        for(int j = 1; j <= n; j++) {
            scanf("%d", &f);
            if(f) {
                for(int y = 1; y <= k; y++)
                    add(co(i, j, y), co(i, j, 0), a);
                if(i < n)//题中说汽车在行驶过程中遇油库则应加满油，所以这只添加0-1的，保证加油
                    add(co(i, j, 0), co(i + 1, j, 1), 0);
                if(j < n)
                    add(co(i, j, 0), co(i, j + 1, 1), 0);
                if(i > 1)
                    add(co(i, j, 0), co(i - 1, j, 1), b);
                if(j > 1)
                    add(co(i, j, 0), co(i, j - 1, 1), b);
            } else {
                for(int y = 1; y <= k; y++)
                    add(co(i, j, y), co(i, j, 0), a + c);
                for(int y = 0; y < k; y++) {
                    if(i < n)
                        add(co(i, j, y), co(i + 1, j, y + 1), 0);
                    if(j < n)
                        add(co(i, j, y), co(i, j + 1, y + 1), 0);
                    if(i > 1)
                        add(co(i, j, y), co(i - 1, j, y + 1), b);
                    if(j > 1)
                        add(co(i, j, y), co(i, j - 1, y + 1), b);
                }
            }
        }
    dijkstra();
    int ans = INF;
    for(int i = 0; i <= k; i++) {
        ans = min(ans, dis[co(n, n, i)]);
    }
    printf("%d\n", ans);
    return 0;
}