AcWing 853. 有边数限制的最短路

问题描述：

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

请你求出 $1$ 号点到 $n$ 号点的最短距离，如果无法从 $1$号点走到 $n$ 号点，则输出 impossible。

输入格式：

第一行包含整数 $n$ ,$m$, $k$。

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

输出格式：

输出一个整数，表示 $1$ 号点到 $n$ 号点的最短距离。

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

数据范围

$1\le n,k\le 500,$
$1\le m\le 10000,$
任意边长的绝对值不超过 $10000$。

输入样例：

3 3 1
1 2 1
2 3 1
1 3 3

输出样例：

3

时间复杂度 $O(mn)$

C++代码：

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 510, M = 10010;

int n, m, k;
int dist[N], bakup[N];

struct Edge
{
    int a, b ,w;
}edges[M];

void bellman_ford()
{
    memset(dist, 0x3f, sizeof dist);
    dist[1] = 0;

    for (int i = 0; i < k; i ++ )
    {
        memcpy(bakup, dist, sizeof dist);
        for(int j = 0; j < m; j ++ )
        {
            int a = edges[j].a, b = edges[j].b, w = edges[j].w;
            dist[b] = min(dist[b], bakup[a] + w);
        }
    }

    if(dist[n] > 0x3f3f3f3f / 2)  printf("impossible");
    else printf("%d\n", dist[n]);
}


int main()
{
    scanf("%d%d%d", &n, &m, &k);

    for (int i = 0; i < m; i ++ )
    {
        int a, b ,w;
        scanf("%d%d%d", &a, &b, &w);
        edges[i] = {a, b, w};
    }

    bellman_ford();

    return 0;
}