在郊区有 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;
}