java---Bellman ford算法---最短路(3)（每日一道算法2022.8.29）

提示：Bellman ford算法一般是在出现负权边时使用，Dijkstra算法是不能在有负权边的时候使用的哦~

题目
给定一个 n 个点 m 条边的有向图，图中可能存在重边和自环， 边权可能为负数
请你求出从 1 号点到 n 号点的最多经过 k 条边的最短距离，如果无法从 1 号点走到 n 号点，输出 impossible

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

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

输入：
3 3 1
1 2 1
2 3 1
1 3 3

输出:
3

public class 最短路_bellman_ford_存在负权边 {
    //初始化，n是点的数量，m是边的数量，k是“最多经过几条边到终点”
    //dist存储起点到当前节点的距离，backup用来备份dist
    //edge存储所有的边和边的权重
    public static int N = 510, large = 0x3f3f3f3f, n, m, k;
    public static int[] dist = new int[N], backup = new int[N];
    public static data[] edge = new data[N];

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        n = in.nextInt(); m = in.nextInt(); k = in.nextInt();

        for (int i = 0; i<m; i++) {
            int x = in.nextInt(), y = in.nextInt(), w = in.nextInt();
            edge[i] = new data(x, y, w);
        }

        bellman_ford();
    }

    public static void bellman_ford() {
        //千万不要忘记dist覆盖正无穷，还有初始节点到自己为0
        Arrays.fill(dist, large);
        dist[1] = 0;

        //题目要求不超过k条边，那就循环k次就行，找不到就是不通咯
        //每次循环的时候记得用backup对dist进行备份
        //这样不会出现假如k是1，a到c是3，a到b是1，b到c是1，然后循环到b点的时候把dist[c]改为2，然后输出2这种无语操作
        for (int i = 0; i<k; i++) {
            backup = Arrays.copyOf(dist, n+1);  //n+1是因为dist中的数据是从1开始存储的

            //m条边所以是m次遍历，每次进行更新距离
            for (int j = 0; j<m; j++) {
                int x = edge[j].x, y = edge[j].y, w = edge[j].w;
                dist[y] = Math.min(dist[y], backup[x] + w);
            }
        }

        //这个地方和Dijkstra不同是因为可能有负权边，就会导致虽然从起点走不到终点，但终点可能连接其他点
        //进行更新的时候会发生正无穷减去一个权重，导致终点虽然无法走通，但终点的dist不是正无穷,那么我们直接比较一个较大的数当作正无穷就行啦
        if (dist[n] > large / 2) System.out.println("impossible");
        else System.out.println(dist[n]);
    }

}

//自定义的数据结构，代表从点x到点y的边的权重为w
class data {
    int x, y, w;
    public data(int x, int y, int w) {
        this.x = x;
        this.y = y;
        this.w = w;
    }
}

很简单，思路就是双循环

声明：算法思路来源为y总，详细请见https://www.acwing.com/
本文仅用作学习记录和交流