BZOJ2763: [JLOI2011]飞行路线

【分析】

【代码】

#include <cstdio>
#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <stack>
#define N 11005
#define M 110005
#define INF 1000000000
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pa;

int read()
{
int x=0,f=1;char ch=getchar();
while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
while(isdigit(ch)){x=x*10+ch-'0';ch=getchar();}
return x*f;
}

int n,m,K,cnt,S,T;
int b[M],p
,nextedge[M],w[M];
int dis
[15];
bool Flag
[15];

void Add(int x,int y,int z)
{
cnt++;
b[cnt]=y;
nextedge[cnt]=p[x];
p[x]=cnt;
w[cnt]=z;
}

void Anode(int x,int y,int z){
Add(x,y,z);Add(y,x,z);
}

void Input_Init()
{
n=read(),m=read(),K=read();
S=read(),T=read();
for(int i=1;i<=m;i++)
{
static int x,y,z;
x=read(),y=read(),z=read();
Anode(x,y,z);
}
}

void spfa()
{
queue<pa>q;
for(int i=0;i<n;i++)
for(int j=0;j<=K;j++) dis[i][j]=INF;
q.push(pa(S,0));dis[S][0]=0;
while(!q.empty())
{
pa k=q.front();q.pop();Flag[k.first][k.second]=0;
for(int i=p[k.first];i;i=nextedge[i])
{
int v=b[i];
if(dis[v][k.second]>dis[k.first][k.second]+w[i])
{
dis[v][k.second]=dis[k.first][k.second]+w[i];
if(!Flag[v][k.second])
{
Flag[v][k.second]=1;
q.push(pa(v,k.second));
}
}
if(k.second<K)
{
if(dis[v][k.second+1]>dis[k.first][k.second])
{
dis[v][k.second+1]=dis[k.first][k.second];
if(!Flag[v][k.second+1])
{
Flag[v][k.second+1]=1;
q.push(pa(v,k.second+1));
}
}
}
}
}
printf("%d\n",dis[T][K]);
}

int main()
{
Input_Init();
spfa();
return 0;
}