行动！行动！

#137. 行动！行动！

描述
提交
自定义测试

【题目描述】：

大CX国的大兵Jack接到一项任务：敌方占领了n座城市（编号0~n-1），有些城市之间有双向道路相连。Jack需要空降在一个城市S，并徒步沿那些道路移动到T城市。虽然Jack每从一个城市到另一个城市都会受伤流血，但大CX国毕竟有着“过硬”的军事实力，它不仅已经算出Jack在每条道路上会损失的血量，还给Jack提供了k个“简易保护罩”，一个保护罩可以让Jack在一条路上的流血量为0。Jack想知道自己最少会流多少血，不过他毕竟是无脑的大兵，需要你的帮助。
【输入描述】：

第一行有三个整数n,m,k，分别表示城市数，道路数和保护罩个数。

第二行有两个整数，S,T。分别表示Jack空降到的城市编号和最终要到的城市。

接下来有m行，每行三个整数a,b,c，表示城市a与城市b之间有一条双向道路。
【输出描述】：

Jack最少要流的血量。
【样例输入】：

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

【样例输出】：

8

【时间限制、数据范围及描述】：

时间：1s 空间：128M

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

#include<cstdio>
#include<cctype>
#include<queue>
#include<cmath>
#include<cstring>
#include<algorithm>
#define rep(i,s,t) for(int i=s;i<=t;i++)
#define dwn(i,s,t) for(int i=s;i>=t;i--)
#define ren for(int i=first[x];i;i=next[i])
using namespace std;
inline int read() {
    int x=0,f=1;char c=getchar();
    for(;!isdigit(c);c=getchar()) if(c=='-') f=-1;
    for(;isdigit(c);c=getchar()) x=x*10+c-'0';
    return x*f;
}
const int maxn=10010;
const int maxm=200010;
struct HeapNode {
    int u,d;
    bool operator < (const HeapNode& ths) const {return d>ths.d;}       
};
int n,m,k,s,t,first[maxn],next[maxm];
int to[maxm],dis[maxm],d[15][maxn],done[maxn*12];
priority_queue<HeapNode> Q;
void spfa() {
    rep(i,0,k) rep(j,0,n-1) d[i][j]=1<<30;
    Q.push((HeapNode){s+n*k,0});d[k][s]=0;
    while(!Q.empty()) {
        if(done[Q.top().u]) {Q.pop();continue;}
        else done[Q.top().u]=1;
        int x=Q.top().u%n,k2=Q.top().u/n;Q.pop();
        ren {
            if(d[k2][to[i]]>d[k2][x]+dis[i]) {
                d[k2][to[i]]=d[k2][x]+dis[i];
                Q.push((HeapNode){to[i]+n*k2,d[k2][to[i]]});  
            }
            if(d[k2-1][to[i]]>d[k2][x]) {
                d[k2-1][to[i]]=d[k2][x];
                Q.push((HeapNode){to[i]+n*k2-n,d[k2-1][to[i]]});   
            }
        }
    }
}
int main() {
    n=read();m=read();k=read();
    s=read();t=read();
    rep(i,1,m) {
        int x=read();to[i]=read();to[i+m]=x;dis[i+m]=dis[i]=read();
        next[i]=first[x];first[x]=i;next[i+m]=first[to[i]];first[to[i]]=i+m;           
    }
    spfa();
    int ans=1<<30;
    rep(i,0,k) ans=min(ans,d[i][t]);
    printf("%d\n",ans);
    return 0;    
}