853. 有边数限制的最短路

题目

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

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

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

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

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

点的编号为 1∼n。

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

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

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

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

思路：

Dijkstra不适用于求带有负权边的图，而bellman_ford对所有边结构体进行遍历更新，遍历k次，可以求得在不超过k条边的情况下，1号点到其他点的最短路径

代码：

#include <iostream>
#include <cstring>

using namespace std;

const int N = 510, M = 10010;

struct{
    int a, b, w;
}edge[M];//点-边-点结构体,存储所有边，方便结构体遍历所有边

int dist[N];//距离数组，每次进行更新
int last[N];//保存上一次更新迭代的结果
int n, m, k;

void bellman_ford(){
    memset(dist, 0x3f, sizeof dist);
    dist[1] = 0;
    
    // bellman_ford的时间复杂度是O(n*n),外层循环是对所有边进行迭代的次数，迭代k次可以求得最多不经过k条边的最短路
    for(int i = 0; i < k; i++){
        // 需要注意，每次对所有边结构体进行迭代，是以上一次迭代的结果对当前最短距离进行更新，因此copy一个距离数组保存上一次迭代的结果
        memcpy(last,dist,sizeof dist);
        // 遍历所有边结构体,更新a到b的最短距离
        for(int j = 0; j < m; j++){
            int a = edge[j].a, b = edge[j].b, w = edge[j].w;
            //如果上一次迭代之后1号点到a的距离+a到b的距离 < 上一次迭代之后1号点到b的距离，则更新1号点到b的最短距离
            dist[b] = min(dist[b], last[a] + w);
        }
    }
    
    // 注意：如果用函数返回值的形式返回最短距离，那么如果刚好最短距离的结果为-1时，在函数外部用是否等于-1进行判断是否有最短路径的话，是不合理的，因此在函数外部之间特判输出对应结果，而不在函数内部用返回值的形式
    // if(dist[n] > 0x3f3f3f3f / 2) return -1;
    // return dist[n];
}

int main()
{
    cin >> n >> m >> k;
    
    for(int i = 0; i < m; i++){
        int a, b, c;
        cin >> a >> b >> c;
        edge[i] = {a, b, c};//点-边-点结构体
    }
    
    bellman_ford();
    
    // 由于某个点到n号点的边权可能为负数，因此在k次迭代过程中可能会把默认初始化为0x3f3f3f3f的dist[n]更新为较小值，k次迭代最多更新k次，而每条边的长度最多为10000，因此dist[n]最多被减少500*10000，因此只需要判断dist[n] > 0x3f3f3f3f/2 即可，这个值依然可以看成是无穷大，可以拿来判断1号点到n号点是否可以到达，无穷大是否有被更新为较小值
    if(dist[n] > 0x3f3f3f3f / 2) puts("impossible");
    else cout << dist[n] << endl;
    
    return 0;
}