Rinne Loves Graph——分层图

传送门
题意
n个城镇，有一些城镇被派兵戒严，求从1到n通过戒严的城镇不超过k次的最短距离。
解法
将在某个城镇通过了多少次戒严的城镇设为一个状态。
d[i][j]：从1到i通过了j次戒严的城镇。
（或者是分层图的做法，建图的时候定义n*k个点，对于加入的一条边，连接能到达的所有状态。）
用dijkstra算法把每个状态看做一个点跑一遍即可。
需要注意的是题中说的通过了一个戒严的城镇是 从一个戒严的点出发到达任意一个点，都会使得次数加1

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
const int maxn = 1e5 + 7;
typedef long long ll;
int n, m, k, a[maxn];
int head[maxn], to[maxn<<1], nex[maxn<<1], edge[maxn<<1], tot;
void init() {
    memset(head, -1, sizeof(head));
    tot = 0;
}
void add(int x, int y, int z) {
    to[tot] = y;
    edge[tot] = z;
    nex[tot] = head[x];
    head[x] = tot++;
}
ll d[maxn][11];
bool final[maxn][11];
struct node {
    int u, k;//当前点，通过了戒严城镇的次数
    ll dis;
    node(int u, int k, ll dis): u(u), k(k), dis(dis) {}
    friend bool operator < (node a, node b) {
        return a.dis > b.dis;
    }
};
void dijkstra(int s) {
    memset(d, INF, sizeof(d));
    memset(final, 0, sizeof(final));
    priority_queue<node> que;
    que.push(node(s, 0, 0)), d[s][0] = 0;//起点如果是戒严的话，没有出发的情况下通过的戒严城镇数还是0
    while (!que.empty()) {
        node p = que.top();
        que.pop();
        if(final[p.u][p.k]) continue;
        final[p.u][p.k] = true;
        for (int i = head[p.u]; ~i; i = nex[i]) {
            int v = to[i];
            int tk = a[p.u] + p.k;//新状态要判断上一个通过的城镇是否是戒严的
            if(tk <= k && !final[v][tk] && d[v][tk] > p.dis + edge[i]) {
                d[v][tk] = p.dis + edge[i];
                que.push(node(v, tk, d[v][tk]));
            }
        }
    }
}
int main()
{
    init();
    scanf("%d%d%d", &n, &m, &k);
    for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
    for (int i = 1, u, v, w; i <= m; i++) {
        scanf("%d%d%d", &u, &v, &w);
        add(u, v, w); add(v, u, w);
    }
    dijkstra(1);
//    for (int i = 1; i <= n; i++) {
//        for (int j = 0; j <= k; j++) {
//            printf("%lld ", d[i][j]);
//        }
//        printf("\n");
//    }
    ll ans = INF;
    for (int i = 0; i <= k; i++)
        ans = min(ans, d[n][i]);
    if(ans == INF) printf("-1\n");
    else printf("%lld\n", ans);
}