BZOJ[2763][JLOI2011]飞行路线 spfa

题目链接http://www.lydsy.com/JudgeOnline/problem.php?id=2763

Description

Alice和Bob现在要乘飞机旅行，他们选择了一家相对便宜的航空公司。该航空公司一共在n个城市设有业务，设这些城市分别标记为0到n-1，一共有m种航线，每种航线连接两个城市，并且航线有一定的价格。Alice和Bob现在要从一个城市沿着航线到达另一个城市，途中可以进行转机。航空公司对他们这次旅行也推出优惠，他们可以免费在最多k种航线上搭乘飞机。那么Alice和Bob这次出行最少花费多少？
Input

数据的第一行有三个整数，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)$

Output

只有一行，包含一个整数，为最少花费。

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

HINT

对于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.

分层图最短路
$dis_{i,j}$表示到第i个点用j次免费的最短路
注意本题为无向图

代码如下：

#include<cstring>
#include<ctype.h>
#include<cstdio>
#include<queue>
#define pii pair<int,int>
#define mp make_pair
#define N 10050
using namespace std;
inline int read(){
    int x=0,f=1;char c;
    do c=getchar(),f=c=='-'?-1:f; while(!isdigit(c));
    do x=(x<<3)+(x<<1)+c-'0',c=getchar(); while(isdigit(c));
    return x*f;
}
queue<pii>q;
int n,m,k,s,t,x,y,z,top,ans;
int fir[N],dis[N][13];
bool b[N][13];
struct Edge{
    int to,nex,k;
    Edge(int _=0,int __=0,int ___=0):to(_),nex(__),k(___){}
}nex[50050*2];
inline void add(int x,int y,int k){
    nex[++top]=Edge(y,fir[x],k);
    fir[x]=top;
}
inline int spfa(){
    while(!q.empty()) q.pop();
    q.push(mp(s,0));dis[s][0]=0;
    while(!q.empty()){
        int x=q.front().first,f=q.front().second;q.pop();
        b[x][f]=false;
        for(int i=fir[x];i;i=nex[i].nex){
            if(dis[nex[i].to][f]>dis[x][f]+nex[i].k){
                dis[nex[i].to][f]=dis[x][f]+nex[i].k;
                if(!b[nex[i].to][f]) b[nex[i].to][f]=true,q.push(mp(nex[i].to,f));
            }
            if(f<k && dis[nex[i].to][f+1]>dis[x][f]){
                dis[nex[i].to][f+1]=dis[x][f];
                if(!b[nex[i].to][f+1]) b[nex[i].to][f+1]=true,q.push(mp(nex[i].to,f+1));
            }
        }
    }
    ans=2147483647;
    for(int i=k;i>=0;i--){
        ans=min(ans,dis[t][i]);
    }
    return ans;
}
int main(){
    n=read();m=read();k=read();
    s=read()+1;t=read()+1;
    for(int i=1;i<=m;i++){
        x=read()+1;y=read()+1;z=read();
        add(x,y,z);add(y,x,z);
    }
    memset(dis,0x7f7f7f7f,sizeof dis);
    printf("%d",spfa());
return 0;
}