所谓分层图就是有多维状态的有边长图,然后比起正常的最短路转移就是多了一种跨维度转移的状态转移。
一般都是舍去某边长度,最多舍多少次(次数较少)。详见代码。
#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; }