现学现用
不过这题能过估计是没满数据。
不然空间要爆。
k层图空间为 :O k*m
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int M = 1e6+7;
int head[M],cnt;
struct EDGE{int to,nxt,val;}ee[M*2];
void add(int x,int y,int z){ee[++cnt].nxt=head[x],ee[cnt].to=y,ee[cnt].val=z,head[x]=cnt;}
int n,m,k,st,ed;
priority_queue<pii,vector<pii>,greater<pii> >q;
int d[M],vs[M];
void dij()
{
memset(d,0x3f,sizeof(d));
memset(vs,0,sizeof(vs));
d[st]=0;
q.push({0,st});
while(q.size())
{
int u=q.top().second;q.pop();
if(vs[u])continue;//第一次取的一定是最优情况
vs[u]=1;
for(int i=head[u];i;i=ee[i].nxt)
{
int v=ee[i].to,w=ee[i].val;
if(d[v]>max(w,d[u]))
d[v]=max(w,d[u]),q.push({d[v],v});
}
}
}
int main()
{
scanf("%d%d%d",&n,&m,&k);
st=1,ed=n;
for(int i=1;i<=m;i++)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
for(int j=0;j<=k;j++)
{
add(u+j*n,v+j*n,w),add(v+j*n,u+j*n,w);
if(j)add(u+(j-1)*n,v+j*n,0),add(v+(j-1)*n,u+j*n,0);
}
}
dij();
int ans=d[ed];
for(int i=1;i<=k;i++)ans=min(ans,d[ed+i*n]);
if(ans>=1e8)printf("-1");
else printf("%d\n",ans);
return 0;
}