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

题目描述
给定一个 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;
}