问题 D: [JLOI2011]飞行路线
时间限制: 1 Sec 内存限制: 128 MB提交: 151 解决: 32
[提交][状态][讨论版]
题目描述
Alice和Bob现在要乘飞机旅行，他们选择了一家相对便宜的航空公司。该航空公司一共在n个城市设有业务，设这些城市分别标记为0到n-1，一共有m种航线，每种航线连接两个城市，并且航线有一定的价格。Alice和Bob现在要从一个城市沿着航线到达另一个城市，途中可以进行转机。航空公司对他们这次旅行也推出优惠，他们可以免费在最多k种航线上搭乘飞机。那么Alice和Bob这次出行最少花费多少？
输入
数据的第一行有三个整数，n,m,k，分别表示城市数，航线数和免费乘坐次数。
第二行有两个整数，s,t，分别表示他们出行的起点城市编号和终点城市编号。(0<=s,t<n)
接下来有m行，每行三个整数，a,b,c，表示存在一种航线，能从城市a到达城市b，或从城市b到达城市a，价格为c。(0<=a,b<n,a与b不相等，0<=c<=1000)
输出
只有一行，包含一个整数，为最少花费。
样例输入
5 6 10 40 1 51 2 52 3 53 4 52 3 30 2 100
样例输出
8
提示
对于30%的数据,2<=n<=50,1<=m<=300,k=0;
对于50%的数据,2<=n<=600,1<=m<=6000,0<=k<=1;
对于100%的数据,2<=n<=10000,1<=m<=50000,0<=k<=10.
就是普通二维spfa，复习下板子。对了，要用堆优化，而且重载时符号别重载反。。
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#define N 10000
using namespace std;
int read()
{
int sum=0,f=1;char x=getchar();
while(x<'0'||x>'9'){if(x=='-')f=-1;x=getchar();}
while(x>='0'&&x<='9'){sum=sum*10+x-'0';x=getchar();}
return sum*f;
}
int n,m,k,adj[N+5],v[N+5][15];
int dis[N+5][15],e,s,t,ans=(1<<30);
struct road
{
int v,next,l;
}lu[N*10+5];
struct node
{
int id,h,dis;
node(){}
node(int x,int y,int z)
{
id=x;h=y;dis=z;
}
bool friend operator <(const node &a,const node &b)
{
return (a.dis!=b.dis)? (a.dis>b.dis):(a.h>b.h);
}
};
priority_queue<node> q;
void add(int u,int v,int l){lu[++e].v=v;lu[e].next=adj[u];lu[e].l=l;adj[u]=e;}
void spfa()
{
dis[s][0]=0;v[s][0]=1;
q.push(node(s,0,0));
while(!q.empty())
{
node w=q.top();q.pop();int tmp=w.h,x=w.id;
for(int i=adj[x];i;i=lu[i].next)
{
int to=lu[i].v;
if(dis[to][tmp]>dis[x][tmp]+lu[i].l)
{
dis[to][tmp]=dis[x][tmp]+lu[i].l;
if(!v[to][tmp])
{
v[to][tmp]=1;
q.push(node(to,tmp,dis[to][tmp]));
}
}
if(tmp<k&&dis[to][tmp+1]>dis[x][tmp])
{
dis[to][tmp+1]=dis[x][tmp];
if(!v[to][tmp+1])
{
v[to][tmp+1]=1;
q.push(node(to,tmp+1,dis[to][tmp+1]));
}
}
}
v[x][tmp]=0;
}
}
int main()
{
n=read();m=read();k=read();s=read();t=read();
int x,y,z;
for(int i=1;i<=m;i++)
{
x=read();y=read();z=read();
add(x,y,z);
add(y,x,z);
}
memset(dis,120,sizeof(dis));
spfa();
for(int i=0;i<=k;i++)
ans=min(ans,dis[t][i]);
cout<<ans;
}