luogu P1948 电话线

#include <iostream>
#include <cstdio>
#include <queue>
using namespace std;
int read()
{
	int x=0,f=1;
	char c=getchar();
	while(c<'0'||c>'9')
	{
		if(c=='-')  f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9')
	{
		x=x*10+c-'0';
		c=getchar();
	}
	return f*x;
}
const int N=1e6+5,M=2e7+5;
int n,m,k,a1,b1,c1,l,r,mid,mi,ans=-1,dis[N];
int had[N],to[M*2],len[M*2],nxt[M*2],p;
bool f[N];
void add(int x,int y,int z)
{
	nxt[++p]=had[x];
	to[p]=y;
	len[p]=z;
	had[x]=p;
}
void addedge(int x,int y,int z)
{
	add(x,y,z);  add(y,x,z);
	for(int i=1;i<=k;i++)
	{
		add(x+n*i,y+n*i,z);  add(y+n*i,x+n*i,z);
		add(x+n*(i-1),y+n*i,0);
		add(y+n*(i-1),x+n*i,0);
	}
}
void dijkstra(int lim)
{
	priority_queue < pair<int,int> > que;
	que.push(make_pair(0,1));
	while(que.size())
	{
		int u=que.top().second;  que.pop();
		if(f[u])  continue;
		f[u]=1;
		for(int i=had[u];i;i=nxt[i])
		{
			int v=to[i];
			if(len[i]>lim)  continue;
			if(dis[v]>dis[u]+len[i])
			{
				dis[v]=dis[u]+len[i];
				que.push(make_pair(-dis[v],v));
			}
		}
	}
}
int main()
{
	n=read();  m=read();  k=read();
	for(int i=1;i<=m;i++)
	{
		a1=read();  b1=read();  c1=read();  r=max(r,c1);
		addedge(a1,b1,c1);
	}
	while(l<=r)
	{
		int mid=(l+r)/2;
		mi=2147483647;  dis[1]=0;
		for(int i=1;i<=n*(k+1);i++)  f[i]=0;
		for(int i=2;i<=n*(k+1);i++)  dis[i]=2147483647;
		dijkstra(mid);
		for(int i=0;i<=k;i++)  mi=min(mi,dis[n*(i+1)]);
		if(mi==2147483647)  l=mid+1;
		else
		{
			ans=mid;
			r=mid-1;
		}
	}
	cout<<ans;
	return 0; 
}