题目:
在郊区有 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
思路:最短路(spfa算法)+dp(dp{i][j]:从第一个点到第i个点,且路径上有j条边的权值改为了0的最小方案数)。dp的状态转化分为:将当前边的权值改为0,和不将当前边的权值改为0(具体写法见代码和注释)。
注意:spfa算法的复杂度最坏为 O(NP),也就是最坏情况下达到1e7。
具体代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int N=2e4+10,M=1e4+10;
int n,p,h;
int e[N],v[N],f[N],ne[N],cur;
void add(int a,int b,int l)//数组模拟邻接表
{
e[++cur]=b,v[cur]=l,ne[cur]=f[a],f[a]=cur;
}
int dp[1010][M];
bool vis[1010];
queue<int >q;
void spfa()
{
while(q.size())q.pop();
memset(vis,false,sizeof vis);
memset(dp,0x3f,sizeof dp);
dp[1][0]=0;
q.push(1);
vis[1]=true;
while(q.size())
{
int start=q.front();
q.pop();
vis[start]=false;
for(int i=f[start];i;i=ne[i])
{
int val=v[i],en=e[i],k=max(dp[start][0],val);
if(dp[en][0]>k)
{
dp[en][0]=k;
if(!vis[en])
{
vis[en]=true;
q.push(en);
}
}
for(int j=1;j<=h;j++)
{
//max(dp[start][j],val)表示当前这条边不该为0
//dp[start][j-1]表示当前这条边改为0
int w=min(max(dp[start][j],val),dp[start][j-1]);
if(w<dp[en][j])
{
dp[en][j]=w;
if(!vis[en])
{
q.push(en);
vis[en]=true;
}
}
}
}
}
}
int main()
{
scanf("%d%d%d",&n,&p,&h);
for(int i=0;i<p;i++)
{
int a,b,l;
scanf("%d%d%d",&a,&b,&l);
add(a,b,l);
add(b,a,l);
}
spfa();
int res=1e9;
for(int i=0;i<=h;i++)
res=min(res,dp[n][i]);
if(res==1e9)
printf("-1\n");
else
printf("%d\n",res);
return 0;
}