Acwing 340 通信线路

题目

在郊区有 N 座通信基站，P 条双向电缆，第 i 条电缆连接基站Ai和Bi
特别地，1 号基站是通信公司的总站，N 号基站位于一座农场中。
现在，农场主希望对通信线路进行升级，其中升级第 i 条电缆需要花费Li
电话公司正在举行优惠活动。
农产主可以指定一条从 1 号基站到 N 号基站的路径，并指定路径上不超过 K 条电缆，由电话公司免费提供升级服务。
农场主只需要支付在该路径上剩余的电缆中，升级价格最贵的那条电缆的花费即可。
求至少用多少钱可以完成升级。
输入格式
第1行：三个整数N，P，K。
第2…P+1行：第 i+1 行包含三个整数Ai,Bi,Li
。
输出格式
包含一个整数表示最少花费。
数据范围
0≤K<N≤1000
1≤P≤10000,
1≤Li≤1000000
输入样例：
5 7 1
1 2 5
3 1 4
2 4 8
3 2 3
5 2 9
3 4 7
4 5 6
输出样例：
4

题解

• 状态：$f[i][j]$ 表示 走到了第$i$个点，经历了$j$条边。
• 方程$$f[v][0]=max(f[u][0],edge[i])$$
  $$f[v][k]=min(f[u][k-1],max(f[u][k]),edge[i])$$

code

#include <bits/stdc++.h> 
using namespace std; 
const int maxn = 1e3 + 100; 
const int maxm = 1e4 + 100; 
const int inf = 0x3f3f3f3f;
typedef long long LL;

template <typename T>
inline void read(T &s) {
    s = 0;
    T w = 1, ch = getchar(); 
    while (!isdigit(ch)) { if (ch == '-') w = -1; ch = getchar(); } 
    while (isdigit(ch))  { s = (s << 1) + (s << 3) + (ch ^ 48); ch = getchar(); } 
    s *= w; 
} 

int n, m, tot, k, ans = 0; 
int lin[maxn], nex[maxm << 1], ver[maxm << 1]; 
int edge[maxm << 1], f[maxn][maxn]; 
bool vis[maxn]; 
queue <int> q;

inline int max(int aa, int bb) { return aa > bb ? aa : bb; }
inline int min(int aa, int bb) { return aa < bb ? aa : bb; }

inline void add(int from, int to, int dis) {
    ver[++tot] = to;
    nex[tot] = lin[from]; 
    edge[tot] = dis;
    lin[from] = tot;
}

void spfa(int s) {
    memset(f, 0x3f, sizeof(f));
    memset(vis, false, sizeof(vis));
    vis[s] = true;
    f[s][0] = 0;
    q.push(s);
    while (!q.empty()) {
        int u = q.front(); q.pop(); vis[u] = false;
        for (int i = lin[u]; i; i = nex[i]) {
            int v = ver[i];
            if (f[v][0] > max(f[u][0], edge[i])) {
                f[v][0] = max(f[u][0], edge[i]);
                if (!vis[v]) { vis[v] = true; q.push(v); }
            }
            for (int j = 1; j <= k; ++j) {
                if (f[v][j] > min( f[u][j - 1], max(f[u][j], edge[i]) )) {
                    f[v][j] = min( f[u][j - 1], max(f[u][j], edge[i]) );
                    if (!vis[v]) { vis[v] = true; q.push(v); }
                }
            }
        }
    }
}

int main() {
    read(n), read(m), read(k);
    for (int i = 1; i <= m; ++i) {
        int x, y, z;
        read(x), read(y), read(z);
        add(x, y, z);
        add(y, x, z);
    }
    spfa(1);
    ans = inf;
    for (int i = 0; i <= k; ++i) {
        ans = min(ans, f[n][i]); 
    }
    if (ans == inf) puts("-1");
    else printf("%lld", ans);
    return 0;
}