leetcode787.k站中转内最便宜的航班

题目

思路
dp[i][k]表示经过k个中转站后到达站i的最低费用
初始化：dp[src][0]~dp[src][k] = 0，其余值为无穷大
状态转移方程：dp[flight[1]][k] = Math.min(dp[flight[1]][k],dp[flight[0]][k-1]+flight[2]);

具体代码

class Solution {
    public int findCheapestPrice(int n, int[][] flights, int src, int dst, int K) {
        int[][] dp = new int[n][K+2];
        for(int i = 0;i < n;i++) Arrays.fill(dp[i],Integer.MAX_VALUE);
        for(int k = 0;k < K+2;k++) dp[src][k] = 0;
        for(int k = 1;k < K+2;k++){
            for(int[] flight: flights){
                if(dp[flight[0]][k-1] != Integer.MAX_VALUE){
                    dp[flight[1]][k] = Math.min(dp[flight[1]][k],dp[flight[0]][k-1]+flight[2]);
                }
            }
        }
        return dp[dst][K+1] == Integer.MAX_VALUE ? -1 : dp[dst][K+1];
    }
}

参考leetcode评论