BZOJ2763: [JLOI2011]飞行路线

spfa 

多记录一层状态  dis[x][i]表示走到x这个点免费走了i次

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<queue>
#include<set>
#include<map>
#include<vector>
#include<complex>
using namespace std;
int dis[11111][11],d[111111][3],head[11111],mark[11111],q[11111];
int n,m,K,st,en,tot=0;
void insert(int x,int y,int v)
{
	d[++tot][0]=y; d[tot][1]=v; d[tot][2]=head[x]; head[x]=tot;
}
void spfa()
{
	memset(dis,127,sizeof(dis));
	for(int i=0; i<=K; i++) dis[st][i]=0;
	int l=1,r=2;   q[1]=st;
	while(l!=r)
	{
		int x=q[l]; l=(l+1)%n; mark[x]=0;
		for(int i=head[x]; i; i=d[i][2])
		{
			int flag=0;
			for(int j=0; j<=K; j++)
			    if(dis[d[i][0]][j]>dis[x][j]+d[i][1])
			        dis[d[i][0]][j]=dis[x][j]+d[i][1],flag=1;
			for(int j=1; j<=K; j++)
			    if(dis[d[i][0]][j]>dis[x][j-1])
			        dis[d[i][0]][j]=dis[x][j-1],flag=1;
			if(flag&&!mark[d[i][0]])
			{
				q[r]=d[i][0];
				r=(r+1)%n;
				mark[d[i][0]]=1;
			}
		}
	}
}
int main()
{
	scanf("%d%d%d%d%d",&n,&m,&K,&st,&en);
	for(int i=1; i<=m; i++)
	{
		int x,y,v; 
		scanf("%d%d%d",&x,&y,&v);
		insert(x,y,v);
		insert(y,x,v);
	}
	spfa();
	printf("%d",dis[en][K]);
	return 0;
}