传送门:通信线路
思路:二分第路径的第k+1数的值,看路径中大于mid的路径条数是否大于k以此来区分出左右区间,在spfa中,把大于mid的看做1,小于mid的看做0,最后dist[n]的值就是大于mid的条数了
代码:
#include<iostream>
#include<cstring>
#include<sstream>
#include<algorithm>
#include<queue>
using namespace std;
typedef pair<int,int>PII;
const int N=2e4+10;
int idx,e[N],ne[N],w[N],h[N];
int n,m,k;
int dist[N];
bool st[N];
int q[N];
void add(int a,int b,int c)
{
e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
int spfa(int x)
{
memset(dist,0x3f,sizeof dist);
dist[1]=0;
memset(st,0,sizeof st);
st[1]=1;
q[0]=1;
int hh=0,tt=1;
while(hh!=tt)
{
int t=q[hh++];
if(hh>=N) hh=0;
st[t]=false;
for(int i=h[t];i!=-1;i=ne[i])
{
int j=e[i];
int p=w[i]>x;
if(dist[j]>dist[t]+p)
{
dist[j]=dist[t]+p;
if(!st[j])
{
q[tt++]=j;
if(tt>=N)
tt=0;
st[j]=true;
}
}
}
}
return dist[n];
}
int main()
{
cin>>n>>m>>k;
memset(h,-1,sizeof h);
for(int i=1;i<=m;i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
add(a,b,c);
add(b,a,c);
}
int l=0,r=1e6+1;
while(l<r)
{
int mid=(l+r)>>1;
if(spfa(mid)<=k) r=mid;
else l=mid+1;
}
if(r==1e6+1) cout<<-1<<endl;
else cout<<r<<endl;
return 0;
}