POJ3662 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。
输出格式
包含一个整数表示最少花费。
若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到N的最短路，然后得到这条路径上最大的边权值，即求的是最短路上的最大边。
然后加上k条免费的部分，这使得每一条边都有0和C(C为原来的边权)两种选择。这样我就可以建分层图，即建立一个新的维度使每条边都有两个分支
然后利用dp的思想，dis[i][j]表示从1到i的最短路中最大的边，并经过了j条权值为0的边。
dis[i][0]的时候就是常规的最短路；其他情况跑k次循环得出走了j条权值为0的边时的值
dis[v][j]=max( dis[v][j],min(dis[x][j-1],max(dis[x][j],c))) 其中x表示当前边的起点，v表示当前边的终点，c表示当前边的权值。dis[x][j-1]表示当前走0边，max(dis[x][j],c)表示走普通边

#include <bits/stdc++.h>
using namespace std;
int n,m,k;
struct stu{
    int next;
    int c;
};
vector<stu> p[20020];
int dis[1010][1010],vis[1010];
void add(int a,int b,int c){
    stu t;
    t.c=c;
    t.next=b;
    p[a].push_back(t);
}
void spfa(int s){
    memset(dis,0x3f,sizeof(dis));
    memset(vis,0,sizeof(vis));
    queue<int> t;
    vis[s]=1;
    dis[s][0]=0;
    t.push(s);
    while(!t.empty()){
        int x=t.front();
        t.pop();
        vis[x]=0;
        for(int i=0;i<p[x].size();i++){
            int v=p[x][i].next;
            int c=p[x][i].c;
            int w=max(dis[x][0],c);
            if(dis[v][0]>w){
                dis[v][0]=w;
                if(!vis[v]){
                    t.push(v);
                    vis[v]=1;
                }
            }
            for(int j=1;j<=k;j++){
                w=min(dis[x][j-1],max(dis[x][j],c));
                if(dis[v][j]>w){
                    dis[v][j]=w;
                    if(!vis[v]){
                        t.push(v);
                        vis[v]=1;
                    }
                }
            }
        }
    }
}
int main()
{
    int a,b,c;
    scanf("%d%d%d",&n,&m,&k);
    for(int i=0;i<m;i++){
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
        add(b,a,c);
    }
    spfa(1);
    int ans=1e9;
    for(int i=0;i<=k;i++)
        ans=min(ans,dis[n][i]);
    if(ans==1e9)
        ans=-1;
    printf("%d\n",ans);
    return 0;
}