【分层图最短路】通信线路

一、前言

曾经有一场比赛分层图最短路的知识点不会导致一道比较简单的题目没写出来。然后今天看到了这道题，刚好巩固一下分层图最短路。

在图论问题中，使用分层图最短路可以有效的解决有选择条件的最短路问题。不同维度之间用有向边单向连接，当状态从低维度通向高纬度时就代表进行了一次条件选择。

二、题面分析

题目链接：通信线路

在郊区有 N 座通信基站，P 条 双向 电缆，第 i 条电缆连接基站 Ai 和 Bi。

特别地，1 号基站是通信公司的总站，N 号基站位于一座农场中。

现在，农场主希望对通信线路进行升级，其中升级第 i 条电缆需要花费 Li。

电话公司正在举行优惠活动。

农产主可以指定一条从 1 号基站到 N 号基站的路径，并指定路径上不超过 K 条电缆，由电话公司免费提供升级服务。

农场主只需要支付在该路径上剩余的电缆中，升级价格最贵的那条电缆的花费即可。

求至少用多少钱可以完成升级。

输入格式
第 1 行：三个整数 N，P，K。

第 2..P+1 行：第 i+1 行包含三个整数 Ai,Bi,Li。

输出格式
包含一个整数表示最少花费。

若 1 号基站与 N 号基站之间不存在路径，则输出 −1。

数据范围
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

分析：在这里我们可以发现我们有k次机会把一条边的边权变为0，然后再找出剩余路中的最长值的最小值。这里如果我们利用分层图建图，然后只要做一次求最短路操作就可以达成目的了。

三、代码分析

//#pragma GCC optimize(2)
#include <bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define INF 0x3f3f3f3f
#define mem(a, b) memset(a, b, sizeof(a))
using namespace std;
typedef pair<int, int> PII;
const int N = 1e6 + 7;
int n, p, k;
int h[N], ne[N], e[N];
int w[N], id = 1;

int dis[N];
bool ch[N];
void add(int a, int b, int c)
{
    ne[id] = h[a];
    h[a] = id;
    e[id] = b;
    w[id] = c;
    id++;
}

void dijkstra()
{
    mem(dis, 0x3f);
    dis[1] = 0;
    priority_queue<PII, vector<PII>, greater<PII>> q;
    q.push({ 0, 1 });
    while (q.size())
    {
        auto te = q.top();
        q.pop();
        int d = te.first;
        int id = te.second;
        if (ch[id])
            continue;
        ch[id] = true;
        for (int i = h[id]; ~i; i = ne[i])
        {
            int j = e[i];
            int z = max(dis[id], w[i]);// 目前这条路的最长边
            if (dis[j] > z)
            {
                dis[j] = z;
                q.push({ dis[j], j });
            }
        }
    }
}

void solve()
{
    mem(h, -1);
    cin >> n >> p >> k;
   for (int i=1;p--;i++)
    {
        int a, b, c;
        cin >> a >> b >> c;
        add(a, b, c);  //第一层
        add(b, a, c);
        for (int j = 1; j <= k; j++)  //升维最短路分层图
        {
            add(a + (j-1) * n, b + j * n, 0);  //从下一层连上来
            add(b + (j-1) * n, a + j * n, 0);
            add(a + j * n, b + j * n, c);   //连接这一层
            add(b + j * n, a + j * n, c);
        }
    }
    for (int i = 1; i <= k; i++)  //将每一层的答案传递到最高层
    {
        add(i * n, (i + 1) * n, 0);
    }
 
    dijkstra();

    if (dis[n * (k + 1)] == INF )  //最短路为无限大则说明最短路不存在
        cout << "-1";
    else
        cout << dis[n * (k + 1)];   //否则输出答案
}

int main()
{
    // std::ios::sync_with_stdio(false);
    // cin.tie(0), cout.tie(0);
    solve();
    return 0;
}