通信线路(多层图法)

在郊区有 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+1层图，方法：

这样就会有对应的在不同层上的相同结点，通过i+k*n，即可得到k层图上的i号结点且编号不与上一层冲突；

建边包括：每层图内建边和相邻的两层图建边，这里让每层图内部的边权和原图一致，然后相邻两层图的对应的相同结点间建立一条权值为0的边，这样即可实现免费的效果，从一层的结点走到下一层图的下一结点，而下一层图上没有当前层的记录，就相当于刚才这一层走的是免费的，bfs时，可以枚举到当前层的下一结点或走到下一层，这样即可枚举所有方案；

注：因为最终要花费k+1贵的，所以用优先队列维护花费的最大值，每次更新时，取比max大的即可。

最后走到k+1层图时，免费的次数已用完，所以维护的最终答案就在k+1层。

完整代码：

#include <iostream>
#include <algorithm>
#include <cstring>
#include <queue>
#include <vector>

#define pll pair<int,int>

using namespace std;

const int maxn=1e6+10,maxm=1e7+10;

int e[maxm],nex[maxm],c[maxm],head[maxn],cnt;
int n,p,k;
int dis[maxn];
bool vis[maxn];

void add(int u,int v,int w)
{
    e[cnt]=v,c[cnt]=w,nex[cnt]=head[u],head[u]=cnt++;
}

int bfs()
{
    priority_queue<pll>q;
    memset(vis,false,sizeof vis);
    memset(dis,0x3f,sizeof dis);
    dis[1]=0;
    q.push({0,1});
    while(q.size()){
        pll t=q.top();
        q.pop();
        int u=t.second;
        if(vis[u]) continue;
        vis[u]=true;
        for(int i=head[u];~i;i=nex[i]){
            int v=e[i],w=max(dis[u],c[i]);
            if(dis[v]>w){
                dis[v]=w;
                q.push({-dis[v],v});
            }
        }
    }
    if(dis[(k+1)*n]==0x3f3f3f3f) return -1;
    return dis[(k+1)*n];
}

int main()
{
    memset(head,-1,sizeof head);
    cin>>n>>p>>k;
    for(int i=1;i<=p;i++){
        int x,y,w;
        cin>>x>>y>>w;
        add(x,y,w);
        add(y,x,w);
        for(int j=1;j<=k;j++){
            add(x+(j-1)*n,y+j*n,0);
            add(y+(j-1)*n,x+j*n,0);
            add(x+j*n,y+j*n,w);
            add(y+j*n,x+j*n,w);
        }
    }
    for(int i=1;i<=k;i++) add(i*n,(i+1)*n,0);
    cout<<bfs()<<endl;
    return 0;
}