有边数限制的最短路问题

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≤10000,
任意边长的绝对值不超过 10000。

输入样例:

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

输出样例:

3

首先我们注意到这里边权可能为负数,那么我们就不能使用之前的dijkstra算法进行求解,为什么呢?
在这里插入图片描述
如上图所示,如果我们使用dijkstra算法,那么我们的步骤如下所示:

  1. 利用1号节点更新与之相连的节点,这里d[2] = 2, d[3] = 5
  2. 然后我们从更新的节点中选取距离最近的点,加入最短路径,即2号点
  3. 然后继续利用2号点更新距离,然后选取最近的点,即4号点,最后继续上述步骤选取五号点
  4. 最终我们得到的序列应该是1->2->4->5,但是很明显1->3->4->5距离更短,因此dijkstra算法失效

这里我们应该多角度考虑问题,而不能每次只选择当前最小,这样的一锤子买卖.

因此,我们引出了bellman-ford算法,代码难度较小。时间复杂度为O(nm)

for n :
	for i in m:(对边(a, b) = w进行松弛操作)
		dist[b] = Math.min(dist[b], backup[a] + w;

上述题目代码实现如下所示:

import java.io.*;
import java.util.*;

public class Main{
    public static int[] backup = new int[100010];
    public static int[] d = new int[100010];
    public static void main(String[] args)throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] lens = br.readLine().split(" ");
        int n = Integer.parseInt(lens[0]);
        int m = Integer.parseInt(lens[1]);
        int k = Integer.parseInt(lens[2]);
        ArrayList<int[]> list = new ArrayList<>();
        for (int i = 0; i < m; i++){
            String[] res =br.readLine().split(" ");
            int a = Integer.parseInt(res[0]);
            int b = Integer.parseInt(res[1]);
            int c = Integer.parseInt(res[2]);
            list.add(new int[]{a, b, c});
        }
        int t = bellman_ford(n, m, k, list);
        if (t == -1) System.out.println("impossible");
        else System.out.println(t);
    }
    public static int bellman_ford(int n, int m, int k, ArrayList<int[]> list){
        Arrays.fill(d, 0x3f3f3f3f);
        d[1] = 0;
        for (int i = 0; i < k; i++){
            backup = Arrays.copyOf(d, n + 1);
            for (int j = 0; j < m; j++){
                int[] x = list.get(j);
                int a = x[0];
                int b = x[1];
                int c = x[2];
                d[b] = Math.min(d[b], backup[a] + c);
            }
        }
        if(d[n] > 0x3f3f3f3f / 2) return -1;
        else return d[n];
    }
}

参考Acwing题解

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值