分层图最短路

我的心愿是世界和平！

一般模型：在图上，有k次机会可以直接通过一条边，问起点与终点之间的最短路径。
dis[i][k]：从起点走到i号点，免费经过了k条边的最短路。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;

#define MAXN 0x3f3f3f3f
int dis[10005][15];//走到i号点，免费经过了j条边的最短路
int head[100005];
bool sign[10005][15];//标记该点是否已经在队列中
struct edge{
    int u,v,w;
    int next;
}e[100005];
struct node{
    int to;
    int k_num;
}nod;//存入队列的结构体
queue<node>q;
int n,m,k,s,t,cnt=1;

void add(int a,int b,int c){
    e[cnt]=((edge){a,b,c,head[a]});
    head[a]=cnt++;
}

void SPFA(){
    memset(dis,MAXN,sizeof(dis));
    memset(sign,0,sizeof(sign));
    q.push((node){s,0});
    dis[s][0]=0;
    sign[s][0]=1;
    while(!q.empty()){
        nod=q.front();
        q.pop();
        sign[nod.to][nod.k_num]=0;
        for(int i=head[nod.to];i!=0;i=e[i].next){
            int nxt=e[i].v;
            if(dis[nxt][nod.k_num]>dis[nod.to][nod.k_num]+e[i].w){
                dis[nxt][nod.k_num]=dis[nod.to][nod.k_num]+e[i].w;
                if(!sign[nxt][nod.k_num]){
                    q.push((node){nxt,nod.k_num});
                    sign[nxt][nod.k_num]=1;
                }
            }
            if(nod.k_num<k){
                if(dis[nxt][nod.k_num+1]>dis[nod.to][nod.k_num])
                {
                    dis[nxt][nod.k_num+1]=dis[nod.to][nod.k_num];
                    if(!sign[nxt][nod.k_num+1])
                    {
                        q.push((node){nxt,nod.k_num+1});
                        sign[nxt][nod.k_num+1]=1;
                    }
                }
            }
        }
    }
}

int main(){
    memset(head,0,sizeof(head));
    scanf("%d%d%d",&n,&m,&k);//输入n点数，m边数,k免费通过机会数
    scanf("%d%d",&s,&t);//输入s起始点，t终点
    for(int i=1;i<=m;i++){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);//输入a，b两点及其距离c
        add(a,b,c);
        add(b,a,c);
    }
    SPFA();
    printf("%d\n",dis[t][k]);
    return 0;
}