acwing 340. 通信线路 分层图解法

现学现用

不过这题能过估计是没满数据。

不然空间要爆。

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;
}