提示: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/
本文仅用作学习记录和交流