南京网络赛 L Magical Girl Haze（优先队列+最短路）

我们设置dis[i][k]表示走到第i号点，免费经过了k条边的最短路。
对于我们当前找到的终点，尝试起点的状态去更新，不选择此条边免费的状态和选择此条边免费的状态，再将这两个状态压入队列去更新可以到达的其他状态。

#include<bits/stdc++.h>
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define ll long long
using namespace std;
const int inf = 0x3f3f3f3f;
const int N = 100005;
int n,m,k;
vector<pair<int,ll> >ve[N];
ll dis[N][15];//设置dis[i][k]表示走到第i号点，免费经过了k条边的最短路。 
bool vis[N][15];
typedef struct Node{
    int a,b;
    ll dis;
    friend bool operator < (const Node &p,const Node &q)
    {
        return p.dis > q.dis;
    }
}Node;
void dijistra()
{
    memset(dis,inf,sizeof(dis));
    memset(vis,false,sizeof(vis));
    priority_queue<Node>que;
    dis[1][0] = 0;
    que.push(Node{1,0,0});
    while(!que.empty())
    {
        Node t = que.top();
        que.pop();
        if(vis[t.a][t.b]) continue;
        vis[t.a][t.b] = true;
        int len = ve[t.a].size();
        for(int i = 0;i < len;++i)
        {
            int x = ve[t.a][i].fi;
            if(dis[x][t.b] > t.dis + ve[t.a][i].se){
                dis[x][t.b] = t.dis + ve[t.a][i].se;
                que.push(Node{x,t.b,dis[x][t.b]});
            }
            if(t.b + 1 <= k){
                if(dis[x][t.b + 1] > t.dis){
                    dis[x][t.b + 1] = t.dis;
                    que.push(Node{x,t.b + 1,dis[x][t.b + 1]});
                }
            }
        }
    }
}

int main()
{
    int t,u,v;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d %d %d",&n,&m,&k);
        for(int i = 1;i <= n;++i)
            ve[i].clear();
        for(int i = 0;i < m;++i)
        {
            int u,v;
            ll c;
            scanf("%d %d %lld",&u,&v,&c);
            ve[u].pb(mp(v,c));
        }
        dijistra();
        printf("%lld\n",dis[n][k]);
    }
    return 0;
}