Leetcode_787_K站中转内最便宜的航班_最短路

最新推荐文章于 2022-04-16 12:05:45 发布

原创最新推荐文章于 2022-04-16 12:05:45 发布 · 151 阅读

0 ·

CC 4.0 BY-SA版权

daily_algorithm 同时被 2 个专栏收录

230 篇文章

订阅专栏

最短路

2 篇文章

订阅专栏

本文介绍了使用动态规划解决旅行商问题的方法，通过模拟深度优先搜索，每步更新dp数组来记录从起始城市到每个城市的最短路径成本。代码实现中，初始化dp数组为极大值，逐步更新并找到从源点到目标点的最短路径，最后返回到达目的地的最小代价。

思想就是模拟一个广搜，广搜一次代表走一步，用dp数组保存到某个节点的最短路长度。

import java.util.Arrays;

class Solution {
    public int findCheapestPrice(int n, int[][] flights, int src, int dst, int k) {
        final int INF = 10000 * 101 + 1;
        int[] dp = new int[n];
        Arrays.fill(dp, INF);
        dp[src] = 0;
        int ans = INF;
        for (int i = 1; i <= k + 1; i++) {
            int[] temp = Arrays.copyOf(dp, n);
            Arrays.fill(temp, INF);
            for (int[] flight : flights) {
                temp[flight[1]] = Math.min(temp[flight[1]], dp[flight[0]] + flight[2]);
            }
            dp = temp;
            ans = Math.min(ans, dp[dst]);
        }
        return ans == INF ? -1 : ans;
    }
}