BZOJ2763 [JLOI2011]飞行路线

题面在这里

水水更健康……

示例程序:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#include<cstdio>
#include<cstring>
#include<algorithm>
#define cl(x,y) memset(x,y,sizeof(x))
using namespace std;

const int maxn=10005,maxe=100005,maxs=maxn*12;
int n,e,K,S,T;
int tot,son[maxe],lnk[maxn],nxt[maxe],w[maxe];
inline void add(int x,int y,int z){
son[++tot]=y;nxt[tot]=lnk[x];lnk[x]=tot;w[tot]=z;
}
struct p{
int x,t;
p() {}
p(int _x,int _t):x(_x),t(_t) {}
}que[maxs];
int dst[maxn][12];
bool vis[maxn][12];
void spfa(){
cl(dst,63);cl(vis,0);
int hed=0,til=1; que[1]=p(S,0);dst[S][0]=0;
while (hed!=til){
int x=que[hed=(hed+1)%maxs].x,t=que[hed].t;
vis[x][t]=0;
for (int j=lnk[x];j;j=nxt[j]){
if (dst[son[j]][t]>dst[x][t]+w[j]){
dst[son[j]][t]=dst[x][t]+w[j];
if (!vis[son[j]][t]) vis[son[j]][t]=1,que[til=(til+1)%maxs]=p(son[j],t);
}
if (t<K&&dst[son[j]][t+1]>dst[x][t]){
dst[son[j]][t+1]=dst[x][t];
if (!vis[son[j]][t+1]) vis[son[j]][t+1]=1,que[til=(til+1)%maxs]=p(son[j],t+1);
}
}
}
}
int main(){
scanf("%d%d%d%d%d",&n,&e,&K,&S,&T);S++,T++;
for (int i=1,x,y,z;i<=e;i++) scanf("%d%d%d",&x,&y,&z),x++,y++,add(x,y,z),add(y,x,z);
spfa(); int ans=0x3f3f3f3f;
for (int i=0;i<=K;i++) ans=min(ans,dst[T][i]);
printf("%d",ans);
return 0;
}