【BZOJ2763】【JLOI2011】飞行路线 分层图

所谓分层图就是有多维状态的有边长图，然后比起正常的最短路转移就是多了一种跨维度转移的状态转移。

一般都是舍去某边长度，最多舍多少次（次数较少）。详见代码。

#include <queue>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 11000
#define M 51000
#define inf 0x3f3f3f3f
using namespace std;
struct KSD
{
	int v,len,next;
}e[M<<1];
int head[N],cnt;
struct Lux
{
	int x,y;
	Lux(int a,int b):x(a),y(b){}
	Lux(){}
};
void add(int u,int v,int len)
{
	++cnt;
	e[cnt].v=v;
	e[cnt].len=len;
	e[cnt].next=head[u];
	head[u]=cnt;
}
int n,m,p,s,t;
int dist[11][N];
bool in[11][N];
queue<Lux>q;
int spfa()
{
	int i,v;
	memset(dist,0x3f,sizeof(dist));
	q.push(Lux(0,s));
	dist[0][s]=0;
	in[0][s]=1;
	while(!q.empty())
	{
		Lux U=q.front();
		q.pop();
		in[U.x][U.y]=0;

		for(i=head[U.y];i;i=e[i].next)
		{
			v=e[i].v;
			if(dist[U.x][v]>dist[U.x][U.y]+e[i].len)
			{
				dist[U.x][v]=dist[U.x][U.y]+e[i].len;
				if(!in[U.x][v])q.push(Lux(U.x,v)),in[U.x][v]=1;
			}
		}
		if(U.x<p)for(i=head[U.y];i;i=e[i].next)
		{
			v=e[i].v;
			if(dist[U.x+1][v]>dist[U.x][U.y])
			{
				dist[U.x+1][v]=dist[U.x][U.y];
				if(!in[U.x+1][v])q.push(Lux(U.x+1,v)),in[U.x+1][v]=1;
			}
		}
	}

	int ret=inf;
	for(i=0;i<=p;i++)ret=min(ret,dist[i][t]);
	return ret;
}

int main()
{
//	freopen("test.in","r",stdin);
	int i,j,k;
	int a,b,c;
	scanf("%d%d%d%d%d",&n,&m,&p,&s,&t);
	for(i=1;i<=m;i++)
	{
		scanf("%d%d%d",&a,&b,&c);
		add(a,b,c);
		add(b,a,c);
	}
	printf("%d\n",spfa());
	return 0;
}