（分层图）洛谷P4568[JLOI2011]飞行路线

洛谷P4568 [JLOI2011]飞行路线

思路：

分层图是对于一些边权的操作，可以将图从二维的引入操作次数$k$作为第三维。
将图复制成$k+1$份，并且上一份的连向下一份的边权值是修改的权值。

代码：

#include<bits/stdc++.h>
#define pii pair<int,int>
#define ll long long
#define cl(x,y) memset(x,y,sizeof(x))
#define loop(x,y,z) for(x=y;x<=z;x++)
#define reve(x,y,z)	for(x=y;x>=z;x--)
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define all(x) x.begin(),x.end()
#define lson x<<1,l,mid
#define rson x<<1|1,mid+1,r
#define INF 1e18
const int N=5e6+10;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
const double eps=1e-8;
const double pi=acos(-1);
using namespace std;
struct edge
{
	int u,v,w;
}e[N];
int head[N],len=0,vis[N],dis[N];
void add(int u,int v,int w)
{
	e[++len]={head[u],v,w};
	head[u]=len;
}
void dij(int s)
{
	priority_queue<pii,vector<pii>,greater<pii> > q;
	q.push(mp(0,s));
	while(!q.empty())
	{
		int i,pre=q.top().se;
		q.pop();
		if(vis[pre])
			continue;
		vis[pre]=1;
		for(i=head[pre];i;i=e[i].u)
		{
			int v=e[i].v,w=e[i].w;
			if(dis[v]>dis[pre]+w)
			{
				dis[v]=dis[pre]+w;
				q.push(mp(dis[v],v));
			}
		}
	}
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	int n,m,i,k,j,s,t;
	cin>>n>>m>>k>>s>>t;
	s++;
	t++;
	for(i=1;i<=n*(k+1);i++)
		dis[i]=inf;
	dis[s]=0;
	for(i=1;i<=m;i++)
	{
		int u,v,w;
		cin>>u>>v>>w;
		u++;
		v++;
		add(u,v,w);
		add(v,u,w);
		for(j=1;j<=k;j++)
		{
			add(u+j*n,v+j*n,w);
			add(v+j*n,u+j*n,w);
			add(u+(j-1)*n,v+j*n,0);
			add(v+(j-1)*n,u+j*n,0);
		}	
	}
	dij(s);
	int ans=dis[t];
	for(i=1;i<=k;i++)
		ans=min(ans,dis[t+i*n]);		
	cout<<ans<<endl;
	return 0;
}