Acwing853.有边数限制的最短路

853.有边数限制的最短路

只能用Bellman-Ford算法做的题：

题目描述

给定一个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 ≤ 10^4，
任意边长的绝对值不超过10000。

输入样式：

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

输出样式：

3

代码：

#include <bits/stdc++.h>
using namespace std;

const int N = 510, M = 10010;

int n, m, k;       // n表示点数，m表示边数
int dist[N], backup[N];        // dist[x]存储1到x的最短路距离, backup存储上一次的状态
 
struct Edge     // 边，a表示出点，b表示入点，w表示边的权重
{
    int a, b, w;
}edges[M];
 
// 求1到n的最短路距离，如果无法从1走到n，则返回-1。
int bellman_ford()
{
    for(int i = 0; i < k; i ++ )
    {
        memcpy(backup, dist, sizeof dist);
        for(int j = 0; j < m; j ++ )		// m条边(a->b,权是w)每条边都询问一下
        {
            int a = edges[j].a, b = edges[j].b, w = edges[j].w;
            dist[b] = min(dist[b], backup[a] + w);
        }
    }
    
 
    if (dist[n] > 0x3f3f3f3f / 2) return -1;    // 0x3f3f3f3f是正无穷，但是由于图中可能有负权边，最后dist[n]的值可能比正无穷小一些，但是它其实还是负无穷的范围(不可达)
    return dist[n];
}

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

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

    int t = bellman_ford();

    if(t == -1) puts("impossible");
    else printf("%d", t);
    
    return 0;
}