[JLOI2011]飞行路线

Description

最短路，给起点和终点，前k个点免费。

Sample Input

5 6 1
0 4
0 1 5
1 2 5
2 3 5
3 4 5
2 3 3
0 2 100

Sample Output

8

#include<cstdio>
#include<cstring>
using namespace std;
struct node
{
int x,y,d,next;
}a[210000];int len,last[110000];
void ins(int x,int y,int d)
{
len++;
a[len].x=x;a[len].y=y;a[len].d=d;
a[len].next=last[x];last[x]=len;
}
struct linode
{
int x,hh;
}list[210000];
int d[11000][20];
bool v[11000][20];
int main()
{
int n,m,K,st,ed,x,y,c;
scanf("%d%d%d",&n,&m,&K);
scanf("%d%d",&st,&ed);st++;ed++;
len=0;memset(last,0,sizeof(last));
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&x,&y,&c);x++;y++;
ins(x,y,c);ins(y,x,c);
}
memset(v,false,sizeof(v));v[st][0]=true;
for(int i=1;i<=n;i++)for(int j=0;j<=K;j++)d[i][j]=999999999;
d[st][0]=0;list[1].x=st;list[1].hh=0;
int head=1,tail=2,ans=999999999;
while(head!=tail)
{
x=list[head].x;int hh=list[head].hh;
if(x==ed){if(d[x][hh]<ans)ans=d[x][hh];}
for(int k=last[x];k;k=a[k].next)
{
y=a[k].y;
if(d[y][hh]>d[x][hh]+a[k].d)
{
d[y][hh]=d[x][hh]+a[k].d;
if(v[y][hh]==false)
{
v[y][hh]=true;
list[tail].x=y;list[tail].hh=hh;
tail++;if(tail==200001)tail=1;
}
}
if(hh+1<=K&&d[y][hh+1]>d[x][hh])
{
d[y][hh+1]=d[x][hh];
if(v[y][hh+1]==false)
{
v[y][hh+1]=true;
list[tail].x=y;list[tail].hh=hh+1;
tail++;if(tail==200001)tail=1;
}
}
}
v[x][hh]=false;
head++;if(head==200001)head=1;
}
printf("%d\n",ans);
return 0;
}