有边数限制的最短路问题

题目链接
给定一个n个点m条边的有向图，图中可能存在重边和自环， 边权可能为负数。

请你求出从1号点到n号点的最多经过k条边的最短距离，如果无法从1号点走到n号点，输出impossible。

注意：图中可能 存在负权回路 。

输入格式
第一行包含三个整数n，m，k。

接下来m行，每行包含三个整数x，y，z，表示存在一条从点x到点y的有向边，边长为z。

输出格式
输出一个整数，表示从1号点到n号点的最多经过k条边的最短距离。

如果不存在满足条件的路径，则输出“impossible”。

数据范围
1≤n,k≤500,
1≤m≤10000,
任意边长的绝对值不超过10000。

输入样例：
3 3 1
1 2 1
2 3 1
1 3 3
输出样例：
3

#include<bits/stdc++.h>
#define inf 0x3f3f3f3f
using namespace std;

const int N = 1e4 + 100;
int n, m, k;
int dis[N], backup[N];


struct node
{
    int a, b, w;
}T[N];

int bellmanford()
{
    memset(dis, inf, sizeof dis);
    dis[1] = 0;
    for(int i = 0; i < k; i ++)
    {
        memcpy(backup, dis, sizeof dis);  //这里备份是为了保证最短路是边数用了k条情况下的最短路。
        for(int j = 0; j < m; j ++)
        {
            int a = T[j].a, b = T[j].b, w = T[j].w;
            dis[b] = min(dis[b], backup[a] + w);
        }
    }
    if(dis[n] > (inf / 2)) return -1;  // 记一下
    else return dis[n];
}

int main()
{
    cin >> n >> m >> k;
    for(int i = 0; i < m; i ++)
        cin >> T[i].a >> T[i].b >> T[i].w;
    
    int t = bellmanford();
    if(t == -1) cout << "impossible";
    else cout << dis[n];
}